将眼科疾病分类的公平性迁移至进展预测,提升弱势群体预测公正性。
TransFair: Transferring Fairness from Ocular Disease Classification to Progression Prediction
- 用公平注意力机制训练公平分类模型,再通过知识蒸馏迁移到进展预测。
- 在2D/3D眼底图像上验证,显著降低不同人群间预测偏差。
- 适合关注医疗AI公平性的研究者与临床应用开发者。
人工智能在自动疾病分类中显著降低了医疗成本并提升了服务可及性,但其公平性问题引发担忧,尤其对弱势群体影响更大。尽管已有方法和大规模数据集缓解分类任务中的群体性能差异,但在疾病进展预测中仍难保证公平性,主要因缺乏多样化的纵向数据用于训练鲁棒且公平的预测模型。本文提出TransFair,旨在将经过公平性增强的眼科疾病分类模型迁移到进展预测任务中,并保持公平性。具体而言,我们利用大量数据训练了一个具备公平注意力机制的高效网络(FairEN)。随后,通过知识蒸馏将该公平分类模型转化为公平进展预测模型,目标是最小化分类与预测模型间的潜在特征距离。我们在二维和三维眼底图像上评估了FairEN与TransFair在公平性增强的分类与进展预测效果。大量实验表明,相比不考虑公平学习的模型,TransFair能有效提升眼科疾病进展预测中的群体公平性。
原文摘要 · Abstract (English)
The use of artificial intelligence (AI) in automated disease classification significantly reduces healthcare costs and improves the accessibility of services. However, this transformation has given rise to concerns about the fairness of AI, which disproportionately affects certain groups, particularly patients from underprivileged populations. Recently, a number of methods and large-scale datasets have been proposed to address group performance disparities. Although these methods have shown effectiveness in disease classification tasks, they may fall short in ensuring fair prediction of disease progression, mainly because of limited longitudinal data with diverse demographics available for training a robust and equitable prediction model. In this paper, we introduce TransFair to enhance demographic fairness in progression prediction for ocular diseases. TransFair aims to transfer a fairness-enhanced disease classification model to the task of progression prediction with fairness preserved. Specifically, we train a fair EfficientNet, termed FairEN, equipped with a fairness-aware attention mechanism using extensive data for ocular disease classification. Subsequently, this fair classification model is adapted to a fair progression prediction model through knowledge distillation, which aims to minimize the latent feature distances between the classification and progression prediction models. We evaluate FairEN and TransFair for fairness-enhanced ocular disease classification and progression prediction using both two-dimensional (2D) and 3D retinal images. Extensive experiments and comparisons with models with and without considering fairness learning show that TransFair effectively enhances demographic equity in predicting ocular disease progression.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。