论文精选

医疗AI模型存在记忆偏差问题

Memorisation bias in medical AI

精选理由

这个研究很实用,告诉你医疗AI可能存在一个你没想到的风险,就是模型可能会记住你自己的历史数据,影响它对你未来健康的判断。

研究指出,医疗AI模型在训练时可能无意记住患者的历史数据,导致在评估患者未来数据时出现偏差。这种现象被称为“记忆偏差”,会降低诊断的准确性和特异性,尤其是在患者健康状况未改变时,模型会过度自信。

原文 · arXiv cs.LG

Memorisation bias in medical AI

Medical AI models hold immense potential to improve patient outcomes, but they are also known to unintentionally memorise individual records from their training datasets. While such memorisation has been linked to targeted privacy attacks, its consequences for clinical deployment, where patients may be assessed by a model that saw their historical data during training, remain poorly understood. Here we show that predictions on a patient's unseen future data can change significantly if a model observed that same patient's anonymised historical data during training, a phenomenon we term "memorisation bias". We demonstrate that this bias exists across diverse data modalities and model architectures, and over prolonged time spans: in some cases, memorisation bias persists on future records acquired decades after the historical records used for training. Moreover, in simulated prospective deployment, memorisation bias has asymmetric effects on the diagnostic accuracy of returning data contributors. When a patient returned with a de novo condition absent from their historical records in the training dataset, diagnostic sensitivity decreased significantly compared to an otherwise identical model not trained on their historical data. Conversely, when their health state was unchanged, both sensitivity and specificity were significantly inflated. Our findings reveal a previously uncharacterised risk in medical AI that arises when a model is deployed on patients who contributed to its training data. This exposes a shortcoming of current model development practice: the de-identification measures designed to protect patients' privacy make it difficult to identify returning contributors and exclude them from the AI-assisted interpretation of their own future data. Mitigating memorisation risks may thus require changes to current model training and deployment protocols.