提出Fair-MoE模型,提升视觉语言模型在医疗场景中的公平性与准确率。
Fair-MoE: Fairness-Oriented Mixture of Experts in Vision-Language Models
- 设计公平导向的专家混合机制,筛选偏见特征并融合多专家信息
- 在哈佛公平医疗数据集上,四个属性的公平性与准确率均显著提升
- 适合关注医疗AI公平性的研究者与从业者使用
公平性是医学伦理的基本原则。视觉语言模型(VLMs)凭借同时利用视觉与语言信息的能力,在医疗领域展现出巨大潜力,可减少对大规模数据集的依赖,并完成复杂任务。然而,当前对VLM应用中公平性的研究仍较有限。若未全面评估公平性,可能引发治疗机会不均等的问题,削弱公众对医学深度学习模型的信任。为建立医疗VLM的可信度,我们提出Fair-MoE,一个专为确保公平性与有效性设计的模型。该模型包含两个核心组件:公平导向的专家混合(FO-MoE)与公平导向损失(FOL)。FO-MoE通过调用多个专业专家,过滤偏见图像块特征,并采用集成方法提取更公平的任务相关特征。FOL是一种新型公平导向损失函数,不仅最小化不同属性间的距离,还优化各类属性分布离散程度的差异。扩展实验表明Fair-MoE在公平性与有效性方面均表现优异。在Harvard-FairVLMed数据集上的测试显示,其在所有四个属性上均实现公平性与准确率的提升。代码将公开可用。
原文摘要 · Abstract (English)
Fairness is a fundamental principle in medical ethics. Vision Language Models (VLMs) have shown significant potential in the medical field due to their ability to leverage both visual and linguistic contexts, reducing the need for large datasets and enabling the performance of complex tasks. However, the exploration of fairness within VLM applications remains limited. Applying VLMs without a comprehensive analysis of fairness could lead to concerns about equal treatment opportunities and diminish public trust in medical deep learning models. To build trust in medical VLMs, we propose Fair-MoE, a model specifically designed to ensure both fairness and effectiveness. Fair-MoE comprises two key components: \textit{the Fairness-Oriented Mixture of Experts (FO-MoE)} and \textit{the Fairness-Oriented Loss (FOL)}. FO-MoE is designed to leverage the expertise of various specialists to filter out biased patch embeddings and use an ensemble approach to extract more equitable information relevant to specific tasks. FOL is a novel fairness-oriented loss function that not only minimizes the distances between different attributes but also optimizes the differences in the dispersion of various attributes' distributions. Extended experiments demonstrate the effectiveness and fairness of Fair-MoE. Tested on the Harvard-FairVLMed dataset, Fair-MoE showed improvements in both fairness and accuracy across all four attributes. Code will be publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。