arXiv:2501.09687cs.LG2025-01NeurIPS被引 16

用不确定性重加权提升抑郁检测公平性,改善多任务学习偏差。

U-Fair: Uncertainty-based Multimodal Multitask Learning for Fairer Depression Detection

  • 基于性别和问卷结构设计不确定性重加权策略。
  • 多任务学习提升准确率与公平性,但存在负迁移风险。
  • 首次将机器学习结果与大规模人群研究对齐,适合医疗AI研究者。

心理健康的机器学习偏见日益成为重要挑战。尽管多任务学习通常优于单任务学习,但其在抑郁检测中的性能与公平性影响仍缺乏研究。本文系统探究了多任务学习在抑郁检测中提升性能与公平性的潜力。提出一种基于性别和不确定性(源于PHQ-8量表结构)的任务重加权方法。结果显示,多任务学习虽整体提升表现与公平性,但存在不一致现象,出现负迁移并缩小帕累托前沿,这在高风险医疗场景中令人担忧。所提方法有效缓解上述问题,提升性能与公平性。各PHQ-8子项任务难度分析结果与最大规模的PHQ-8子项区分能力研究一致,首次为机器学习发现提供了与大规模人群研究相印证的实证依据。

原文摘要 · Abstract (English)

Machine learning bias in mental health is becoming an increasingly pertinent challenge. Despite promising efforts indicating that multitask approaches often work better than unitask approaches, there is minimal work investigating the impact of multitask learning on performance and fairness in depression detection nor leveraged it to achieve fairer prediction outcomes. In this work, we undertake a systematic investigation of using a multitask approach to improve performance and fairness for depression detection. We propose a novel gender-based task-reweighting method using uncertainty grounded in how the PHQ-8 questionnaire is structured. Our results indicate that, although a multitask approach improves performance and fairness compared to a unitask approach, the results are not always consistent and we see evidence of negative transfer and a reduction in the Pareto frontier, which is concerning given the high-stake healthcare setting. Our proposed approach of gender-based reweighting with uncertainty improves performance and fairness and alleviates both challenges to a certain extent. Our findings on each PHQ-8 subitem task difficulty are also in agreement with the largest study conducted on the PHQ-8 subitem discrimination capacity, thus providing the very first tangible evidence linking ML findings with large-scale empirical population studies conducted on the PHQ-8.

抑郁检测多任务学习公平性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。