LLM2D
基于临床灵感的分层多标签分类方法:带有惩罚性损失函数的胸部X光分类
Clinically-Inspired Hierarchical Multi-Label Classification of Chest X-rays with a Penalty-Based Loss Function
作者: Mehrdad Asadi, Komi Sodok\'e, Ian J. Gerard, Marta Kersten-Oertel
发布日期: 2/7/2025
arXiv ID: oai:arXiv.org:2502.03591v1

摘要

arXiv:2502.03591v1 提交类型: cross 摘要: 在本文中,我们提出了一种新的方法,用于增强临床可解释性的多标签胸片(X-ray)图像分类,同时保持单一模型、单一运行的训练管道。我们利用CheXpert数据集和VisualCheXbert衍生的标签,引入层次标签分组以捕捉诊断之间的临床意义关系。为实现这一目标,我们设计了一个自定义的层次二元交叉熵(HBCE)损失函数,该函数使用固定的或数据驱动的惩罚类型来强制执行标签依赖性。我们的模型在测试集上达到了0.903的平均受试者操作特征曲线下的面积(AUROC)。此外,我们还提供了可视化解释和不确定性估计,以进一步增强模型的可解释性。所有代码、模型配置和实验细节均公开提供。