提升CT影像诊断公平性与鲁棒性,兼顾不同医院和人群表现。
Towards Fair and Robust Volumetric CT Classification via KL-Regularised Group Distributionally Robust Optimisation
- 用KL正则化分组分布鲁棒优化,动态提升弱势群体表现。
- 在新冠分类任务中F1达0.835,优于现有最佳结果5.9个百分点。
- 显著改善女性鳞状细胞癌等罕见组合的识别准确率。
基于胸部计算机断层扫描(CT)的自动化诊断在临床应用中面临两大挑战:跨采集机构的数据分布偏移以及不同人口学亚群间的性能差异。本文针对两个互补任务:多中心CT体积数据的二分类新冠诊断(任务1),以及带性别公平约束的四类肺部病理识别(任务2)。采用轻量级MobileViT-XXS切片编码器与双层SliceTransformer聚合器进行体素推理,并使用KL正则化分组分布鲁棒优化(Group DRO)目标函数训练,自适应提升表现较差的采集中心与人口学子群体权重。相比标准Group DRO,KL惩罚项防止了组权重坍缩,实现最差情况保护与平均性能之间的稳定平衡。在任务2中,以性别为分组粒度,直接缓解女性鳞状细胞癌等严重低频组合的识别困境。任务1中最佳配置获得0.835的挑战F1,超越已有最优结果5.9个百分点;任务2中α=0.5的Group DRO达成0.815的均性别宏F1,优于最佳基准11.1个百分点,女性鳞状细胞癌的F1相较焦点损失基线提升17.4个百分点。
原文摘要 · Abstract (English)
Automated diagnosis from chest computed tomography (CT) scans faces two persistent challenges in clinical deployment: distribution shift across acquisition sites and performance disparity across demographic subgroups. We address both simultaneously across two complementary tasks: binary COVID-19 classification from multi-site CT volumes (Task 1) and four-class lung pathology recognition with gender-based fairness constraints (Task 2). Our framework combines a lightweight MobileViT-XXS slice encoder with a two-layer SliceTransformer aggregator for volumetric reasoning, and trains with a KL-regularised Group Distributionally Robust Optimisation (Group DRO) objective that adaptively upweights underperforming acquisition centres and demographic subgroups. Unlike standard Group DRO, the KL penalty prevents group weight collapse, providing a stable balance between worst-case protection and average performance. For Task 2, we define groups at the granularity of gender class, directly targeting severely underrepresented combinations such as female Squamous cell carcinoma. On Task 1, our best configuration achieves a challenge F1 of 0.835, surpassing the best published challenge entry by +5.9. On Task 2, Group DRO with α = 0.5 achieves a mean per-gender macro F1 of 0.815, outperforming the best challenge entry by +11.1 pp and improving Female Squamous F1 by +17.4 over the Focal Loss baseline.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。