Private and interpretable clinical prediction with quantum-inspired tensor train models
cs.LG, cs.CR, quant-ph
Submitted: 2026-02-05
Updated: 2026-08-27
Comments: 23 pages, 3 figures, 7 tables. The code for the experiments is publicly available at https://github.com/joserapa98/tns4loris. v2: Published version. OpenReview: https://openreview.net/forum?id=QtG3fC1v5t
Journal ref: Transactions on Machine Learning Research (2026)
Code: https://github.com/joserapa98/tns4loris
License: http://creativecommons.org/licenses/by/4.0/
The gist: 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
Terminology
Abstract
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.
Sources
- Deep Learning with Differential Privacy
- Hacking Smart Machines with Smarter Ones: How to Extract Meaningful Data from Machine Learning Classifiers
- Reconstructing Training Data with Informed Adversaries
- Differentially Private Empirical Risk Minimization
- Matrix Product States and Projected Entangled Pair States: Concepts, Symmetries, and Theorems
- Issues Encountered Deploying Differential Privacy
- MemGuard: Defending against Black-Box Membership Inference Attacks via Adversarial Examples
- Differentially Private Low-Rank Adaptation of Large Language Model Using Federated Learning
- A Matrix Product State Model for Simultaneous Classification and Generation
- Tensorizing Neural Networks
- Exponential Machines
- A Practical Introduction to Tensor Networks: Matrix Product States and Projected Entangled Pair States
- Reconstructing Training Data From Real World Models Trained with Transfer Learning
- Semi-supervised Knowledge Transfer for Deep Learning from Private Training Data
- TensorKrowch: Smooth integration of tensor networks in machine learning
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Scikit-learn: Machine Learning in Python
- Privacy-preserving machine learning with tensor networks
- Matrix Product State Representations
- Membership Privacy for Machine Learning Models Through Knowledge Transfer
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks