用张量分解技术让医疗模型既隐私安全又可解释。
Private and interpretable clinical prediction with quantum-inspired tensor train models
- 将模型转为张量列车结构,隐藏参数防止信息泄露。
- 35人小队在数百至数千人训练集中仍可被精准识别。
- 适合注重可解释性与隐私的临床模型部署场景。
公开的临床机器学习模型存在未被重视的隐私风险:其参数或输出可能被用来恢复训练数据中的患者信息。以美国政府网站上公开的免疫治疗反应预测逻辑回归(LR)模型LORIS为例,我们通过其公开接口成功恢复模型参数并高置信度识别出训练队列。我们设计了三种攻击级别下的群体成员推断攻击(二值黑盒、连续黑盒、白盒),应用于同一任务的LR与浅层神经网络(NN)。结果表明,即使仅35名患者组成的队列,在数百至数千人的训练集中也可被可靠识别,且交叉验证等常规做法反而加剧风险。为此,我们提出一种受量子启发的防御方法,将离散化模型张量化为张量列车(TT),该表示能有效混淆参数、保持精度,并提供接近实用差分隐私水平的黑盒保护。同时保留逻辑回归的可解释性,还通过高效计算边缘与条件分布扩展分析能力,使黑盒模型如NN也能实现更丰富的可解释分析。结果表明,张量化是一种实用、事后适用的私密、可解释且高效的临床预测工具。
原文摘要 · Abstract (English)
Publicly available clinical machine learning models pose an underappreciated privacy risk: their parameters or outputs can be exploited to recover information from patients whose data were used during training. Moreover, this risk is exacerbated by models such as logistic regression (LR), which are typically preferred in clinical settings for their transparency. To assess this empirically, we attack LORIS, a publicly available LR model for immunotherapy response prediction hosted on a U.S. government website. From evaluations through its public interface, we recover the model parameters and identify the training cohort with high confidence. More broadly, we design cohort-level membership inference attacks under three levels of adversarial access---binary black-box, continuous black-box, and white-box---and apply them to both LR models and shallow neural networks (NNs) trained on the same task. Our results reveal that even a cohort of 35 patients can be reliably identified within training sets of hundreds to thousands, and that common practices such as cross-validation amplify rather than mitigate this risk. To address these vulnerabilities, we propose a quantum-inspired defense based on tensorizing discretized models into tensor trains (TTs). This representation obfuscates model parameters and preserves accuracy, while offering black-box protection comparable to practical Differential Privacy baselines. Additionally, the TT representations retain LR interpretability and extend it through efficient computation of marginal and conditional distributions, enabling this richer analysis also for black-box models such as NNs. Our results establish tensorization as a practical, post-hoc tool for private, interpretable, and effective clinical prediction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。