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