Gaussian certified unlearning in high dimensions: A hypothesis testing approach
Aaradhya Pandey, Arnab Auddy, Haolin Zou, Arian Maleki, Sanjeev Kulkarni
We introduce the canonical dimension free notion of certifiability suitable to high dimensions and show its utility via a Newton based unlearning algorithm
Abstract
Machine unlearning seeks to efficiently remove the influence of selected data while preserving generalization. Significant progress has been made in low dimensions, where the dimension of the parameter $p$ is much smaller than the sample size $n$, but high dimensions, including proportional regimes $p \sim n$, pose serious theoretical challenges as standard optimization assumptions of $\Omega(1)$ strong convexity and $O(1)$ smoothness of the per-example loss $f$ rarely hold simultaneously in proportional regimes $p\sim n$. In this work, we introduce $\varepsilon$-Gaussian certifiability, a canonical and robust notion well-suited to high-dimensional regimes, that optimally captures a broad class of noise adding mechanisms. Then we theoretically analyze the performance of a widely used unlearning algorithm based on one step of the Newton method in the high-dimensional setting described above. Our analysis shows that a single Newton step, followed by a well-calibrated Gaussian noise, is sufficient to achieve both privacy and accuracy in this setting. This result stands in sharp contrast to the only prior work that analyzes machine unlearning in high dimensions \citet{zou2025certified}, which relaxes some of the standard optimization assumptions for high-dimensional applicability, but operates under the notion of $\varepsilon$-certifiability. That work concludes %that a single Newton step is insufficient even for removing a single data point, and that at least two steps are required to ensure both privacy and accuracy. Our result leads us to conclude that the discrepancy in the number of steps arises because of the sub optimality of the notion of $\varepsilon$-certifiability and its incompatibility with noise adding mechanisms, which $\varepsilon$-Gaussian certifiability is able to overcome optimally.
Analyzes machine unlearning in high dimensions showing single noisy Newton step with Gaussian noise suffices for privacy-accuracy.
- Introduces epsilon-Gaussian certifiability concept optimal for high-dimensional unlearning with noise-adding mechanisms
- Proves one-step Gaussian-Newton method achieves both privacy and accuracy in proportional regime p~n
- Shows improved gap over prior work requiring two steps through better notion of certifiability
- Newton method
- Gaussian noise addition
- Hypothesis testing framework
- Differential privacy
Linear Hawkes process assumption does not guarantee non-negativity of intensity function
from the paperComputational complexity scales cubically with dimensionality, suitable for moderate dimensions up to few hundred
from the paper
Analyze beyond generalized linear models and high-dimensional regimes
from the paperExtend to non-convex loss functions for deep neural networks
from the paperInvestigate approximate second-order methods like conjugate gradient to reduce Hessian computation cost
from the paperStrengthen scaling for multiple simultaneous deletions from m^4=o(n) to m^3=o(n)
from the paperStudy distributional unlearning including class or concept unlearning
from the paperHandle continuous unlearning requests in online setting
from the paper
Author keywords
- Machine unlearning in high dimensions
- Proportional asymptotics
- High dimensional statistical theory
- Privacy–accuracy tradeoff
- Hypothesis testing
- Gaussian noise calibration
- Newton method
Related orals
LLM Fingerprinting via Semantically Conditioned Watermarks
Introduces semantically conditioned watermarks for robust and stealthy LLM fingerprinting robust to deployment scenarios.
Steering the Herd: A Framework for LLM-based Control of Social Learning
Framework studying strategic control of social learning by algorithmic information mediators with theoretical analysis and LLM-based simulations.
Every Language Model Has a Forgery-Resistant Signature
Ellipse signatures function as forgery-resistant model output identifiers based on high-dimensional geometric constraints.
Differentially Private Domain Discovery
WGM-based methods provide efficient domain discovery with near-optimal guarantees for missing mass on Zipfian data.
What's In My Human Feedback? Learning Interpretable Descriptions of Preference Data
WIMHF uses sparse autoencoders to extract human-interpretable features from preference data, enabling better understanding and curation of human feedback.