Add main.tex
Browse files
main.tex
ADDED
|
@@ -0,0 +1,619 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
\documentclass[final]{neurips_2026}
|
| 2 |
+
|
| 3 |
+
\usepackage[utf8]{inputenc}
|
| 4 |
+
\usepackage[T1]{fontenc}
|
| 5 |
+
\usepackage{hyperref}
|
| 6 |
+
\usepackage{url}
|
| 7 |
+
\usepackage{booktabs}
|
| 8 |
+
\usepackage{amsfonts}
|
| 9 |
+
\usepackage{amsmath}
|
| 10 |
+
\usepackage{amssymb}
|
| 11 |
+
\usepackage{amsthm}
|
| 12 |
+
\usepackage{nicefrac}
|
| 13 |
+
\usepackage{microtype}
|
| 14 |
+
\usepackage{graphicx}
|
| 15 |
+
\usepackage{xcolor}
|
| 16 |
+
\usepackage{enumitem}
|
| 17 |
+
\usepackage{algorithm}
|
| 18 |
+
\usepackage{algorithmic}
|
| 19 |
+
\usepackage{mathtools}
|
| 20 |
+
\usepackage{thm-restate}
|
| 21 |
+
|
| 22 |
+
% Theorem environments
|
| 23 |
+
\newtheorem{theorem}{Theorem}
|
| 24 |
+
\newtheorem{lemma}[theorem]{Lemma}
|
| 25 |
+
\newtheorem{proposition}[theorem]{Proposition}
|
| 26 |
+
\newtheorem{corollary}[theorem]{Corollary}
|
| 27 |
+
\newtheorem{definition}[theorem]{Definition}
|
| 28 |
+
\newtheorem{assumption}[theorem]{Assumption}
|
| 29 |
+
\newtheorem{remark}[theorem]{Remark}
|
| 30 |
+
\newtheorem{example}[theorem]{Example}
|
| 31 |
+
|
| 32 |
+
% Shortcuts
|
| 33 |
+
\newcommand{\R}{\mathbb{R}}
|
| 34 |
+
\newcommand{\E}{\mathbb{E}}
|
| 35 |
+
\newcommand{\KL}{\mathrm{KL}}
|
| 36 |
+
\newcommand{\ELBO}{\mathcal{L}}
|
| 37 |
+
\newcommand{\setminus}{\smallsetminus}
|
| 38 |
+
\newcommand{\cN}{\mathcal{N}}
|
| 39 |
+
\newcommand{\cG}{\mathcal{G}}
|
| 40 |
+
\newcommand{\bU}{\mathbf{U}}
|
| 41 |
+
\newcommand{\bV}{\mathbf{V}}
|
| 42 |
+
\newcommand{\bX}{\mathbf{X}}
|
| 43 |
+
\newcommand{\blambda}{\boldsymbol{\lambda}}
|
| 44 |
+
\DeclareMathOperator{\diag}{diag}
|
| 45 |
+
\DeclareMathOperator{\Gam}{Gamma}
|
| 46 |
+
\DeclareMathOperator{\Poi}{Poisson}
|
| 47 |
+
\DeclareMathOperator{\Mult}{Mult}
|
| 48 |
+
\DeclareMathOperator{\Tr}{Tr}
|
| 49 |
+
\DeclareMathOperator*{\argmin}{arg\,min}
|
| 50 |
+
\DeclareMathOperator*{\argmax}{arg\,max}
|
| 51 |
+
|
| 52 |
+
\title{When Is Local Unlearning Possible?\\ A Weighted Dobrushin Theory for Variational Inference}
|
| 53 |
+
|
| 54 |
+
\author{%
|
| 55 |
+
Anonymous Author(s)
|
| 56 |
+
}
|
| 57 |
+
|
| 58 |
+
\begin{document}
|
| 59 |
+
|
| 60 |
+
\maketitle
|
| 61 |
+
|
| 62 |
+
\begin{abstract}
|
| 63 |
+
We study the problem of efficiently removing the influence of a single data point from a fitted variational inference model without full retraining.
|
| 64 |
+
We develop a \emph{weighted Dobrushin theory} that provides sufficient conditions under which the deletion influence decays exponentially with graph distance from the deleted observation, enabling local unlearning.
|
| 65 |
+
The key quantity is the \emph{weighted interaction mass}~$\chi(z)$, which captures how strongly each observation couples the variational parameters of its endpoints through the coordinate-ascent variational inference (CAVI) fixed-point map.
|
| 66 |
+
When $\chi(z) < 1$ in a suitable spectral sense, we prove that the variational parameter change due to deletion decays as $O(e^{-\mu \cdot d})$, where $d$ is the graph distance and $\mu > 0$ depends on the spectral gap of the weighted Dobrushin matrix.
|
| 67 |
+
We instantiate the theory for Gamma--Poisson, Gaussian--Gaussian, and Gaussian--Gamma matrix factorization models, derive closed-form expressions for the interaction constants, and validate the theory on synthetic and real-world recommendation datasets.
|
| 68 |
+
Local radius-$R$ unlearning achieves up to $3\times$ speedup over exact retraining at $R=1$ with relative error below $8\%$, and the error decreases exponentially with~$R$.
|
| 69 |
+
\end{abstract}
|
| 70 |
+
|
| 71 |
+
%% ============================================================
|
| 72 |
+
\section{Introduction}
|
| 73 |
+
\label{sec:intro}
|
| 74 |
+
|
| 75 |
+
Machine unlearning---the problem of removing the influence of a specific training data point from a learned model---has attracted significant attention due to privacy regulations such as the GDPR ``right to be forgotten''~\citep{bourtoule2021machine,cao2015towards,ginart2019making}.
|
| 76 |
+
The na\"ive approach of retraining from scratch is computationally prohibitive for large models, motivating the development of \emph{approximate unlearning} methods that efficiently update the model to approximate the retrained solution~\citep{guo2020certified,neel2021descent,izzo2021approximate,sekhari2021remember}.
|
| 77 |
+
|
| 78 |
+
Most existing unlearning methods operate \emph{globally}: they modify all model parameters to account for the deleted data point.
|
| 79 |
+
For large-scale models, even a single Newton step or influence-function correction requires touching the entire parameter vector~\citep{koh2017understanding}.
|
| 80 |
+
This raises a natural question:
|
| 81 |
+
|
| 82 |
+
\begin{quote}
|
| 83 |
+
\emph{When is it possible to unlearn locally---by updating only a small neighborhood of parameters near the deleted observation, while leaving the rest unchanged?}
|
| 84 |
+
\end{quote}
|
| 85 |
+
|
| 86 |
+
We answer this question for \emph{variational inference} (VI) models fitted by coordinate-ascent variational inference (CAVI).
|
| 87 |
+
CAVI optimizes a mean-field variational objective by iteratively updating one block of variational parameters at a time, holding the others fixed.
|
| 88 |
+
The structure of the CAVI update map induces a natural notion of \emph{interaction}: block~$u$'s update depends on block~$v$ only if $u$ and $v$ share an observed data point.
|
| 89 |
+
In the bipartite user--item models common in recommendation and topic modeling, this interaction structure is determined by the observation graph.
|
| 90 |
+
|
| 91 |
+
\paragraph{Contributions.}
|
| 92 |
+
\begin{enumerate}[nosep,leftmargin=*]
|
| 93 |
+
\item We develop a \emph{weighted Dobrushin theory} for CAVI fixed points (Section~\ref{sec:theory}). We define a weighted interaction matrix whose entries capture the sensitivity of each block's CAVI update to perturbations in its neighbors, weighted by the data values. When the spectral radius of this matrix is less than~1, the CAVI fixed point is unique and deletion influence decays exponentially with graph distance.
|
| 94 |
+
|
| 95 |
+
\item We derive \emph{closed-form interaction constants} for Gamma--Poisson matrix factorization (Section~\ref{sec:gamma_poisson}), the workhorse model for count data in recommendation and text mining. The constants depend on the digamma function of the shape parameters, the rate parameters, and the observed counts, providing an interpretable proxy for deletion difficulty.
|
| 96 |
+
|
| 97 |
+
\item We extend the framework to Gaussian--Gaussian and Gaussian--Gamma matrix factorization (Section~\ref{sec:extensions}), showing that the locality phenomenon is not specific to the Poisson--Gamma conjugate pair.
|
| 98 |
+
|
| 99 |
+
\item We validate the theory on synthetic data across three graph families and three model families, and on real-world data from Last.fm and MovieLens (Section~\ref{sec:experiments}). Local $R=1$ unlearning achieves $3\times$ speedup with $<8\%$ relative error; the error decays exponentially with radius.
|
| 100 |
+
\end{enumerate}
|
| 101 |
+
|
| 102 |
+
\paragraph{Related work.}
|
| 103 |
+
\emph{Machine unlearning} methods can be grouped into exact methods based on data partitioning~\citep{bourtoule2021machine,chen2022graph}, approximate methods based on influence functions or Newton steps~\citep{guo2020certified,izzo2021approximate,neel2021descent}, and certified methods with formal deletion guarantees~\citep{sekhari2021remember,ullah2021machine}.
|
| 104 |
+
Our work is closest to the influence-function approach but exploits the graph structure of CAVI to restrict updates to a local neighborhood.
|
| 105 |
+
|
| 106 |
+
The \emph{Dobrushin uniqueness condition}~\citep{dobrushin1968description,dobrushin1970prescribing} is a classical tool in statistical mechanics for establishing uniqueness of Gibbs measures and decay of correlations.
|
| 107 |
+
It has been applied to analyze convergence of belief propagation~\citep{tatikonda2002loopy}, Glauber dynamics~\citep{hayes2006simple}, and localized inference~\citep{loh2017efficient}.
|
| 108 |
+
Our contribution is to adapt the Dobrushin condition to the \emph{variational} inference setting, where the ``measure'' is the CAVI fixed point rather than a Gibbs measure, and to derive \emph{weighted} versions that account for heterogeneous data values.
|
| 109 |
+
|
| 110 |
+
\emph{Gamma--Poisson matrix factorization}~\citep{gopalan2015scalable,zhou2012beta,cemgil2009bayesian} is a widely used model for count data.
|
| 111 |
+
The augmented representation with auxiliary multinomial variables admits closed-form CAVI updates~\citep{gopalan2015scalable}, making it an ideal testbed for our theory.
|
| 112 |
+
|
| 113 |
+
%% ============================================================
|
| 114 |
+
\section{Problem Setup}
|
| 115 |
+
\label{sec:setup}
|
| 116 |
+
|
| 117 |
+
\paragraph{Bipartite observation model.}
|
| 118 |
+
Consider a bipartite graph $G = (U, V, E)$ with user nodes $U = \{1, \ldots, N\}$, item nodes $V = \{1, \ldots, M\}$, and observed edges $E \subseteq U \times V$.
|
| 119 |
+
For each edge $(i,j) \in E$, we observe a data value $x_{ij}$.
|
| 120 |
+
We denote the neighborhood of user $i$ as $\Omega_i = \{j : (i,j) \in E\}$ and of item $j$ as $\Omega_j = \{i : (i,j) \in E\}$.
|
| 121 |
+
|
| 122 |
+
\paragraph{Latent variable model.}
|
| 123 |
+
Each user $i$ has a latent factor $U_i \in \R^K$ and each item $j$ has a latent factor $V_j \in \R^K$.
|
| 124 |
+
The data $x_{ij}$ is generated from a likelihood that depends on $U_i$ and $V_j$.
|
| 125 |
+
We place independent priors on $U_i$ and $V_j$.
|
| 126 |
+
|
| 127 |
+
\paragraph{Mean-field variational inference.}
|
| 128 |
+
We approximate the posterior $p(U, V \mid X)$ with a factorized distribution:
|
| 129 |
+
\begin{equation}
|
| 130 |
+
q(U, V; \blambda) = \prod_{i=1}^N q(U_i; \lambda^U_i) \prod_{j=1}^M q(V_j; \lambda^V_j),
|
| 131 |
+
\end{equation}
|
| 132 |
+
where $\blambda = (\lambda^U_1, \ldots, \lambda^U_N, \lambda^V_1, \ldots, \lambda^V_M)$ collects all variational parameters.
|
| 133 |
+
We write $\blambda_u$ for the parameter block of node $u$ (either a user or item).
|
| 134 |
+
The ELBO is:
|
| 135 |
+
\begin{equation}
|
| 136 |
+
\ELBO(\blambda) = \E_q[\log p(X, U, V)] - \E_q[\log q(U, V; \blambda)].
|
| 137 |
+
\end{equation}
|
| 138 |
+
|
| 139 |
+
\paragraph{CAVI.}
|
| 140 |
+
Coordinate-ascent variational inference updates each block $\blambda_u$ by maximizing $\ELBO$ with all other blocks fixed:
|
| 141 |
+
\begin{equation}
|
| 142 |
+
\label{eq:cavi}
|
| 143 |
+
\blambda_u^{(t+1)} = F_u(\blambda_{-u}^{(t)}),
|
| 144 |
+
\end{equation}
|
| 145 |
+
where $F_u$ is the CAVI update map for block $u$.
|
| 146 |
+
A fixed point $\blambda^\star$ satisfies $\blambda_u^\star = F_u(\blambda_{-u}^\star)$ for all $u$.
|
| 147 |
+
|
| 148 |
+
\paragraph{Deletion and unlearning.}
|
| 149 |
+
Given a deletion request for observation $z = (i, j, x_{ij})$, the goal is to compute the variational parameters $\blambda^{\setminus z}$ that would result from fitting the model on $E \setminus \{z\}$.
|
| 150 |
+
\emph{Exact deletion} retrains from scratch (or from a warm start).
|
| 151 |
+
\emph{Local radius-$R$ unlearning} defines a seed set $S(z) = \{U_i, V_j\}$ and updates only blocks within graph distance $R$ of $S(z)$, holding all other blocks at their full-data values $\blambda^\star$.
|
| 152 |
+
|
| 153 |
+
%% ============================================================
|
| 154 |
+
\section{Weighted Dobrushin Theory for CAVI}
|
| 155 |
+
\label{sec:theory}
|
| 156 |
+
|
| 157 |
+
\subsection{The CAVI Interaction Matrix}
|
| 158 |
+
|
| 159 |
+
The key object is the \emph{CAVI interaction matrix} $\mathbf{C}$, an $(N+M) \times (N+M)$ matrix whose entries measure how strongly block $u$'s CAVI update depends on block $v$.
|
| 160 |
+
|
| 161 |
+
\begin{definition}[CAVI interaction coefficient]
|
| 162 |
+
\label{def:interaction}
|
| 163 |
+
For blocks $u, v$ in the variational model, define:
|
| 164 |
+
\begin{equation}
|
| 165 |
+
C_{uv} = \sup_{\blambda} \left\| \frac{\partial F_u(\blambda_{-u})}{\partial \blambda_v} \right\|,
|
| 166 |
+
\end{equation}
|
| 167 |
+
where the norm is the operator norm between the parameter spaces of blocks $v$ and $u$, and the supremum is taken over a suitable domain of~$\blambda$.
|
| 168 |
+
\end{definition}
|
| 169 |
+
|
| 170 |
+
In the bipartite setting, $C_{uv} = 0$ unless $u$ and $v$ share an edge.
|
| 171 |
+
For a user block $U_i$, its update depends only on the item blocks $V_j$ for $j \in \Omega_i$, and vice versa.
|
| 172 |
+
|
| 173 |
+
\begin{definition}[Weighted Dobrushin condition]
|
| 174 |
+
\label{def:dobrushin}
|
| 175 |
+
The \emph{weighted Dobrushin condition} holds if the spectral radius of $\mathbf{C}$ satisfies:
|
| 176 |
+
\begin{equation}
|
| 177 |
+
\rho(\mathbf{C}) < 1.
|
| 178 |
+
\end{equation}
|
| 179 |
+
Equivalently, there exist positive weights $w_u > 0$ such that for all $u$:
|
| 180 |
+
\begin{equation}
|
| 181 |
+
\label{eq:weighted_dobrushin}
|
| 182 |
+
\sum_{v \neq u} C_{uv} \frac{w_v}{w_u} < 1.
|
| 183 |
+
\end{equation}
|
| 184 |
+
\end{definition}
|
| 185 |
+
|
| 186 |
+
\subsection{Main Theorem: Exponential Decay of Deletion Influence}
|
| 187 |
+
|
| 188 |
+
\begin{theorem}[Deletion influence decay]
|
| 189 |
+
\label{thm:main}
|
| 190 |
+
Suppose the weighted Dobrushin condition (Definition~\ref{def:dobrushin}) holds with $\rho(\mathbf{C}) = 1 - \delta$ for some $\delta > 0$.
|
| 191 |
+
Let $\blambda^\star$ be the CAVI fixed point on the full data, and $\blambda^{\setminus z}$ be the fixed point after deleting observation $z = (i,j,x_{ij})$.
|
| 192 |
+
Then for any block $u$ at graph distance $d(u, S(z)) = r$ from the seed set $S(z) = \{U_i, V_j\}$:
|
| 193 |
+
\begin{equation}
|
| 194 |
+
\|\blambda_u^\star - \blambda_u^{\setminus z}\| \leq C_0 \cdot (1 - \delta)^r,
|
| 195 |
+
\end{equation}
|
| 196 |
+
where $C_0$ depends on the magnitude of the deleted observation's contribution to the seed blocks' CAVI updates.
|
| 197 |
+
\end{theorem}
|
| 198 |
+
|
| 199 |
+
The proof (Appendix~\ref{app:proof_main}) proceeds by analyzing the perturbation of the CAVI fixed-point equation.
|
| 200 |
+
Deleting $z$ changes the update map only for the seed blocks $U_i$ and $V_j$.
|
| 201 |
+
The change propagates through the interaction matrix: each hop multiplies by $\mathbf{C}$, and since $\rho(\mathbf{C}) < 1$, the perturbation contracts geometrically.
|
| 202 |
+
|
| 203 |
+
\begin{corollary}[Local unlearning error bound]
|
| 204 |
+
\label{cor:local}
|
| 205 |
+
Under the conditions of Theorem~\ref{thm:main}, local radius-$R$ unlearning satisfies:
|
| 206 |
+
\begin{equation}
|
| 207 |
+
\|\blambda^{(R)}_{\mathrm{local}} - \blambda^{\setminus z}\| \leq C_0 \cdot \frac{(1-\delta)^{R+1}}{1 - (1-\delta)} = \frac{C_0}{\delta} (1-\delta)^{R+1}.
|
| 208 |
+
\end{equation}
|
| 209 |
+
\end{corollary}
|
| 210 |
+
|
| 211 |
+
This shows that local unlearning error decreases \emph{exponentially} with radius $R$, with rate determined by the spectral gap $\delta$.
|
| 212 |
+
|
| 213 |
+
\subsection{Weighted Interaction Mass}
|
| 214 |
+
|
| 215 |
+
For a specific deletion $z = (i, j, x_{ij})$, we define the \emph{weighted interaction mass} at the seed nodes:
|
| 216 |
+
|
| 217 |
+
\begin{definition}[Weighted interaction mass]
|
| 218 |
+
\label{def:chi}
|
| 219 |
+
\begin{equation}
|
| 220 |
+
\chi_i = \sum_{j' \in \Omega_i} C_{U_i, V_{j'}}, \qquad
|
| 221 |
+
\widetilde{\chi}_j = \sum_{i' \in \Omega_j} C_{V_j, U_{i'}}, \qquad
|
| 222 |
+
\chi(z) = \max\{\chi_i, \widetilde{\chi}_j\}.
|
| 223 |
+
\end{equation}
|
| 224 |
+
\end{definition}
|
| 225 |
+
|
| 226 |
+
If $\chi(z) < 1$, the Dobrushin condition holds \emph{at least locally} around the seed, and the influence of deleting $z$ decays.
|
| 227 |
+
The quantity $\chi(z)$ serves as a per-deletion proxy for locality strength: deletions with low $\chi$ are ``easy'' (strongly local), while those with high $\chi$ are ``hard'' (potentially non-local).
|
| 228 |
+
|
| 229 |
+
%% ============================================================
|
| 230 |
+
\section{Gamma--Poisson Matrix Factorization}
|
| 231 |
+
\label{sec:gamma_poisson}
|
| 232 |
+
|
| 233 |
+
\subsection{Model}
|
| 234 |
+
|
| 235 |
+
The augmented Gamma--Poisson model~\citep{gopalan2015scalable} is:
|
| 236 |
+
\begin{align}
|
| 237 |
+
U_{ik} &\sim \Gam(a_0, b_0), \quad V_{jk} \sim \Gam(c_0, d_0), \quad k = 1, \ldots, K, \\
|
| 238 |
+
x_{ij} \mid U_i, V_j &\sim \Poi\!\left(\rho_x \sum_{k=1}^K U_{ik} V_{jk}\right).
|
| 239 |
+
\end{align}
|
| 240 |
+
Using the Poisson--multinomial augmentation, each count $x_{ij}$ is split into component-wise counts $z_{ijk}$ with $\sum_k z_{ijk} = x_{ij}$ and:
|
| 241 |
+
\begin{equation}
|
| 242 |
+
(z_{ij1}, \ldots, z_{ijK}) \mid U_i, V_j \sim \Mult\!\left(x_{ij};\; \frac{U_{ik} V_{jk}}{\sum_\ell U_{i\ell} V_{j\ell}}\right).
|
| 243 |
+
\end{equation}
|
| 244 |
+
|
| 245 |
+
\subsection{CAVI Updates}
|
| 246 |
+
|
| 247 |
+
The mean-field variational family is $q(U_{ik}) = \Gam(a_{ik}, b_{ik})$, $q(V_{jk}) = \Gam(c_{jk}, d_{jk})$.
|
| 248 |
+
The responsibilities are:
|
| 249 |
+
\begin{equation}
|
| 250 |
+
r_{ijk} = \frac{\exp(\psi(a_{ik}) - \log b_{ik} + \psi(c_{jk}) - \log d_{jk})}{\sum_\ell \exp(\psi(a_{i\ell}) - \log b_{i\ell} + \psi(c_{j\ell}) - \log d_{j\ell})},
|
| 251 |
+
\end{equation}
|
| 252 |
+
where $\psi$ is the digamma function.
|
| 253 |
+
The CAVI updates are:
|
| 254 |
+
\begin{align}
|
| 255 |
+
a_{ik} &= a_0 + \sum_{j \in \Omega_i} x_{ij} r_{ijk}, & b_{ik} &= b_0 + \sum_{j \in \Omega_i} \frac{c_{jk}}{d_{jk}}, \\
|
| 256 |
+
c_{jk} &= c_0 + \sum_{i \in \Omega_j} x_{ij} r_{ijk}, & d_{jk} &= d_0 + \sum_{i \in \Omega_j} \frac{a_{ik}}{b_{ik}}.
|
| 257 |
+
\end{align}
|
| 258 |
+
|
| 259 |
+
\subsection{Interaction Constants}
|
| 260 |
+
|
| 261 |
+
\begin{proposition}[Interaction constants for Gamma--Poisson]
|
| 262 |
+
\label{prop:pg_constants}
|
| 263 |
+
The CAVI interaction coefficients for the Gamma--Poisson model satisfy:
|
| 264 |
+
\begin{equation}
|
| 265 |
+
C_{U_i, V_j} \leq C_x \cdot x_{ij} + C_0,
|
| 266 |
+
\end{equation}
|
| 267 |
+
where:
|
| 268 |
+
\begin{equation}
|
| 269 |
+
C_x = \tfrac{1}{2}\psi'(c_{\min}) + \tfrac{1}{2 d_{\min}}, \qquad
|
| 270 |
+
C_0 = \tfrac{1}{d_{\min}} + \tfrac{c_{\max}}{d_{\min}^2},
|
| 271 |
+
\end{equation}
|
| 272 |
+
and $c_{\min}, d_{\min}, c_{\max}$ are bounds on the variational parameters at the fixed point.
|
| 273 |
+
Similarly, $C_{V_j, U_i} \leq \widetilde{C}_x \cdot x_{ij} + \widetilde{C}_0$ with:
|
| 274 |
+
\begin{equation}
|
| 275 |
+
\widetilde{C}_x = \tfrac{1}{2}\psi'(a_{\min}) + \tfrac{1}{2 b_{\min}}, \qquad
|
| 276 |
+
\widetilde{C}_0 = \tfrac{1}{b_{\min}} + \tfrac{a_{\max}}{b_{\min}^2}.
|
| 277 |
+
\end{equation}
|
| 278 |
+
\end{proposition}
|
| 279 |
+
|
| 280 |
+
The proof (Appendix~\ref{app:proof_constants}) differentiates the CAVI update with respect to the neighbor's parameters and bounds the resulting Jacobian.
|
| 281 |
+
The $C_x \cdot x_{ij}$ term captures the count-dependent coupling (higher counts create stronger interaction), while $C_0$ captures the baseline coupling through the rate-parameter updates.
|
| 282 |
+
|
| 283 |
+
The weighted interaction mass at the seed of deletion $z = (i,j,x_{ij})$ is:
|
| 284 |
+
\begin{equation}
|
| 285 |
+
\chi_i = \sum_{j' \in \Omega_i} (C_x \cdot x_{ij'} + C_0), \qquad
|
| 286 |
+
\widetilde{\chi}_j = \sum_{i' \in \Omega_j} (\widetilde{C}_x \cdot x_{i'j} + \widetilde{C}_0).
|
| 287 |
+
\end{equation}
|
| 288 |
+
|
| 289 |
+
\paragraph{Interpretation.}
|
| 290 |
+
The Dobrushin condition $\chi(z) < 1$ is easier to satisfy when: (i) the graph is sparse (few neighbors), (ii) counts are low, (iii) priors are strong (large $b_0, d_0$ push $b_{\min}, d_{\min}$ away from zero), and (iv) the rank $K$ is small (fewer components to couple).
|
| 291 |
+
This provides actionable guidance: locality is strongest for sparse, low-count, strongly regularized models.
|
| 292 |
+
|
| 293 |
+
%% ============================================================
|
| 294 |
+
\section{Extensions to Other Model Families}
|
| 295 |
+
\label{sec:extensions}
|
| 296 |
+
|
| 297 |
+
\subsection{Gaussian--Gaussian Matrix Factorization}
|
| 298 |
+
|
| 299 |
+
For $U_{ik} \sim \cN(0, \sigma_U^2)$, $V_{jk} \sim \cN(0, \sigma_V^2)$, and $x_{ij} \mid U_i, V_j \sim \cN(U_i^\top V_j, \sigma_x^2)$, the mean-field variational family is $q(U_{ik}) = \cN(m^U_{ik}, s^U_{ik})$, $q(V_{jk}) = \cN(m^V_{jk}, s^V_{jk})$.
|
| 300 |
+
|
| 301 |
+
The interaction proxy for this model is:
|
| 302 |
+
\begin{equation}
|
| 303 |
+
\chi_i^{GG} = \sum_{j \in \Omega_i} \frac{1}{\sigma_x^2} \E_q[\|V_j\|^2], \qquad
|
| 304 |
+
\widetilde{\chi}_j^{GG} = \sum_{i \in \Omega_j} \frac{1}{\sigma_x^2} \E_q[\|U_i\|^2].
|
| 305 |
+
\end{equation}
|
| 306 |
+
|
| 307 |
+
Locality is stronger when observation noise $\sigma_x^2$ is large (weaker coupling) and priors are strong (small $\sigma_U^2, \sigma_V^2$).
|
| 308 |
+
|
| 309 |
+
\subsection{Gaussian--Gamma MAP}
|
| 310 |
+
|
| 311 |
+
For nonnegative factors with Gamma priors and Gaussian likelihood, we optimize the MAP objective:
|
| 312 |
+
\begin{equation}
|
| 313 |
+
\max_{U \geq 0, V \geq 0} \left\{ -\frac{1}{2\sigma_x^2} \sum_{(i,j) \in E} (x_{ij} - U_i^\top V_j)^2 + \sum_{i,k} \log p(U_{ik}) + \sum_{j,k} \log p(V_{jk}) \right\}.
|
| 314 |
+
\end{equation}
|
| 315 |
+
|
| 316 |
+
Since this is non-convex, the ``exact deletion'' solution depends on the optimization path.
|
| 317 |
+
The locality theory applies approximately: we use the same interaction proxy but note that the guarantees are weaker because the fixed point may not be unique.
|
| 318 |
+
|
| 319 |
+
%% ============================================================
|
| 320 |
+
%% EXPERIMENTS (imported from separate file)
|
| 321 |
+
%% ============================================================
|
| 322 |
+
\input{latex_results_section}
|
| 323 |
+
|
| 324 |
+
%% ============================================================
|
| 325 |
+
\section{Conclusion}
|
| 326 |
+
\label{sec:conclusion}
|
| 327 |
+
|
| 328 |
+
We have developed a weighted Dobrushin theory that characterizes when local unlearning is possible for variational inference models.
|
| 329 |
+
The theory identifies the weighted interaction mass $\chi(z)$ as the key quantity controlling locality: when $\chi(z) < 1$, deletion influence decays exponentially with graph distance.
|
| 330 |
+
Experiments on synthetic and real data confirm that:
|
| 331 |
+
(i) influence decay is robust across graph types and model families,
|
| 332 |
+
(ii) local radius-$R$ unlearning error decreases exponentially with $R$,
|
| 333 |
+
(iii) the speedup is most pronounced for small $R$ on sparse graphs,
|
| 334 |
+
and (iv) conjugate VI models exhibit stronger locality than non-convex MAP alternatives.
|
| 335 |
+
|
| 336 |
+
The main limitation is that the theoretical proxy $\chi(z)$ does not tightly predict per-deletion difficulty in practice---it provides a valid sufficient condition but is conservative.
|
| 337 |
+
Future work could tighten the bound using spectral methods or develop data-dependent proxies.
|
| 338 |
+
Another direction is scaling the experiments to large graphs ($N \gg 10^4$) where the runtime benefits of local unlearning are most significant.
|
| 339 |
+
|
| 340 |
+
%% ============================================================
|
| 341 |
+
\bibliographystyle{plainnat}
|
| 342 |
+
\begin{thebibliography}{99}
|
| 343 |
+
|
| 344 |
+
\bibitem[Bourtoule et~al.(2021)]{bourtoule2021machine}
|
| 345 |
+
L.~Bourtoule, V.~Chandrasekaran, C.A. Choquette-Choo, H.~Jia, A.~Travers, B.~Zhang, D.~Lie, and N.~Papernot.
|
| 346 |
+
\newblock Machine unlearning.
|
| 347 |
+
\newblock In \emph{IEEE S\&P}, 2021.
|
| 348 |
+
|
| 349 |
+
\bibitem[Cao and Yang(2015)]{cao2015towards}
|
| 350 |
+
Y.~Cao and J.~Yang.
|
| 351 |
+
\newblock Towards making systems forget with machine unlearning.
|
| 352 |
+
\newblock In \emph{IEEE S\&P}, 2015.
|
| 353 |
+
|
| 354 |
+
\bibitem[Cemgil(2009)]{cemgil2009bayesian}
|
| 355 |
+
A.T. Cemgil.
|
| 356 |
+
\newblock Bayesian inference for nonnegative matrix factorisation models.
|
| 357 |
+
\newblock \emph{Computational Intelligence and Neuroscience}, 2009.
|
| 358 |
+
|
| 359 |
+
\bibitem[Chen et~al.(2022)]{chen2022graph}
|
| 360 |
+
M.~Chen, Z.~Zhang, T.~Wang, M.~Backes, M.~Humbert, and Y.~Zhang.
|
| 361 |
+
\newblock Graph unlearning.
|
| 362 |
+
\newblock In \emph{CCS}, 2022.
|
| 363 |
+
|
| 364 |
+
\bibitem[Dobrushin(1968)]{dobrushin1968description}
|
| 365 |
+
R.L. Dobrushin.
|
| 366 |
+
\newblock Description of a random field by means of conditional probabilities and the conditions governing its regularity.
|
| 367 |
+
\newblock \emph{Theory of Probability and its Applications}, 13(2):197--224, 1968.
|
| 368 |
+
|
| 369 |
+
\bibitem[Dobrushin(1970)]{dobrushin1970prescribing}
|
| 370 |
+
R.L. Dobrushin.
|
| 371 |
+
\newblock Prescribing a system of random variables by conditional distributions.
|
| 372 |
+
\newblock \emph{Theory of Probability and its Applications}, 15(3):458--486, 1970.
|
| 373 |
+
|
| 374 |
+
\bibitem[Ginart et~al.(2019)]{ginart2019making}
|
| 375 |
+
A.~Ginart, M.~Guan, G.~Valiant, and J.~Zou.
|
| 376 |
+
\newblock Making {AI} forget you: Data deletion in machine learning.
|
| 377 |
+
\newblock In \emph{NeurIPS}, 2019.
|
| 378 |
+
|
| 379 |
+
\bibitem[Gopalan et~al.(2015)]{gopalan2015scalable}
|
| 380 |
+
P.~Gopalan, J.M. Hofman, and D.M. Blei.
|
| 381 |
+
\newblock Scalable recommendation with hierarchical {P}oisson factorization.
|
| 382 |
+
\newblock In \emph{UAI}, 2015.
|
| 383 |
+
|
| 384 |
+
\bibitem[Guo et~al.(2020)]{guo2020certified}
|
| 385 |
+
C.~Guo, T.~Goldstein, A.~Hannun, and L.~van~der~Maaten.
|
| 386 |
+
\newblock Certified data removal from machine learning models.
|
| 387 |
+
\newblock In \emph{ICML}, 2020.
|
| 388 |
+
|
| 389 |
+
\bibitem[Hayes and Sinclair(2006)]{hayes2006simple}
|
| 390 |
+
T.P. Hayes and A.~Sinclair.
|
| 391 |
+
\newblock A simple condition implying rapid mixing of single-site dynamics on spin systems.
|
| 392 |
+
\newblock In \emph{FOCS}, 2006.
|
| 393 |
+
|
| 394 |
+
\bibitem[Izzo et~al.(2021)]{izzo2021approximate}
|
| 395 |
+
Z.~Izzo, M.A. Smart, K.~Chaudhuri, and J.~Zou.
|
| 396 |
+
\newblock Approximate data deletion from machine learning models.
|
| 397 |
+
\newblock In \emph{AISTATS}, 2021.
|
| 398 |
+
|
| 399 |
+
\bibitem[Koh and Liang(2017)]{koh2017understanding}
|
| 400 |
+
P.W. Koh and P.~Liang.
|
| 401 |
+
\newblock Understanding black-box predictions via influence functions.
|
| 402 |
+
\newblock In \emph{ICML}, 2017.
|
| 403 |
+
|
| 404 |
+
\bibitem[Loh and Wainwright(2017)]{loh2017efficient}
|
| 405 |
+
P.-L. Loh and M.J. Wainwright.
|
| 406 |
+
\newblock Efficient localized inference for large graphical models.
|
| 407 |
+
\newblock \emph{arXiv:1710.10404}, 2017.
|
| 408 |
+
|
| 409 |
+
\bibitem[Neel et~al.(2021)]{neel2021descent}
|
| 410 |
+
S.~Neel, A.~Roth, and S.~Sharifi-Malvajerdi.
|
| 411 |
+
\newblock Descent-to-delete: Gradient-based methods for machine unlearning.
|
| 412 |
+
\newblock In \emph{ALT}, 2021.
|
| 413 |
+
|
| 414 |
+
\bibitem[Sekhari et~al.(2021)]{sekhari2021remember}
|
| 415 |
+
A.~Sekhari, J.~Acharya, G.~Kamath, and A.T. Suresh.
|
| 416 |
+
\newblock Remember what you want to forget: Algorithms for machine unlearning.
|
| 417 |
+
\newblock In \emph{NeurIPS}, 2021.
|
| 418 |
+
|
| 419 |
+
\bibitem[Tatikonda and Jordan(2002)]{tatikonda2002loopy}
|
| 420 |
+
S.~Tatikonda and M.I. Jordan.
|
| 421 |
+
\newblock Loopy belief propagation and {G}ibbs measures.
|
| 422 |
+
\newblock In \emph{UAI}, 2002.
|
| 423 |
+
|
| 424 |
+
\bibitem[Ullah et~al.(2021)]{ullah2021machine}
|
| 425 |
+
E.~Ullah, T.~Mai, A.~Rao, R.A. Rossi, and R.~Arora.
|
| 426 |
+
\newblock Machine unlearning via algorithmic stability.
|
| 427 |
+
\newblock In \emph{COLT}, 2021.
|
| 428 |
+
|
| 429 |
+
\bibitem[Zhou et~al.(2012)]{zhou2012beta}
|
| 430 |
+
M.~Zhou, L.~Hannah, D.~Dunson, and L.~Carin.
|
| 431 |
+
\newblock Beta-negative binomial process and {P}oisson factor analysis.
|
| 432 |
+
\newblock In \emph{AISTATS}, 2012.
|
| 433 |
+
|
| 434 |
+
\end{thebibliography}
|
| 435 |
+
|
| 436 |
+
%% ============================================================
|
| 437 |
+
%% APPENDIX
|
| 438 |
+
%% ============================================================
|
| 439 |
+
\newpage
|
| 440 |
+
\appendix
|
| 441 |
+
|
| 442 |
+
\section{Proof of Theorem~\ref{thm:main} (Deletion Influence Decay)}
|
| 443 |
+
\label{app:proof_main}
|
| 444 |
+
|
| 445 |
+
\begin{proof}
|
| 446 |
+
Let $\blambda^\star$ denote the CAVI fixed point on the full data and $\blambda^{\setminus z}$ the fixed point on the data with observation $z = (i,j,x_{ij})$ removed.
|
| 447 |
+
Let $F$ and $F^{\setminus z}$ denote the full-data and deletion CAVI maps, respectively.
|
| 448 |
+
|
| 449 |
+
The two fixed points satisfy:
|
| 450 |
+
\begin{align}
|
| 451 |
+
\blambda_u^\star &= F_u(\blambda_{-u}^\star), \\
|
| 452 |
+
\blambda_u^{\setminus z} &= F_u^{\setminus z}(\blambda_{-u}^{\setminus z}).
|
| 453 |
+
\end{align}
|
| 454 |
+
|
| 455 |
+
For blocks $u \notin S(z) = \{U_i, V_j\}$, the CAVI update is unchanged: $F_u = F_u^{\setminus z}$.
|
| 456 |
+
Therefore:
|
| 457 |
+
\begin{equation}
|
| 458 |
+
\blambda_u^\star - \blambda_u^{\setminus z} = F_u(\blambda_{-u}^\star) - F_u(\blambda_{-u}^{\setminus z}).
|
| 459 |
+
\end{equation}
|
| 460 |
+
|
| 461 |
+
By the mean value theorem and the definition of the interaction coefficients:
|
| 462 |
+
\begin{equation}
|
| 463 |
+
\|\blambda_u^\star - \blambda_u^{\setminus z}\| \leq \sum_{v} C_{uv} \|\blambda_v^\star - \blambda_v^{\setminus z}\|.
|
| 464 |
+
\end{equation}
|
| 465 |
+
|
| 466 |
+
Define $\Delta_u = \|\blambda_u^\star - \blambda_u^{\setminus z}\|$ and $\boldsymbol{\Delta} = (\Delta_u)_u$.
|
| 467 |
+
Then:
|
| 468 |
+
\begin{equation}
|
| 469 |
+
\boldsymbol{\Delta} \leq \mathbf{C} \boldsymbol{\Delta} + \boldsymbol{\epsilon},
|
| 470 |
+
\end{equation}
|
| 471 |
+
where $\boldsymbol{\epsilon}$ is nonzero only at the seed blocks (capturing the direct effect of removing $z$).
|
| 472 |
+
|
| 473 |
+
Since $\rho(\mathbf{C}) < 1$, the matrix $(\mathbf{I} - \mathbf{C})$ is invertible with:
|
| 474 |
+
\begin{equation}
|
| 475 |
+
\boldsymbol{\Delta} \leq (\mathbf{I} - \mathbf{C})^{-1} \boldsymbol{\epsilon} = \sum_{t=0}^{\infty} \mathbf{C}^t \boldsymbol{\epsilon}.
|
| 476 |
+
\end{equation}
|
| 477 |
+
|
| 478 |
+
The $t$-th power $\mathbf{C}^t$ has the property that $[\mathbf{C}^t]_{uv} = 0$ unless there is a path of length $t$ from $v$ to $u$ in the interaction graph.
|
| 479 |
+
Since $\boldsymbol{\epsilon}$ is supported only on $S(z)$, the contribution to block $u$ at distance $r$ from $S(z)$ first appears at $t = r$:
|
| 480 |
+
\begin{equation}
|
| 481 |
+
\Delta_u \leq \|\boldsymbol{\epsilon}\|_\infty \sum_{t=r}^{\infty} \|\mathbf{C}^t\|_\infty \leq \|\boldsymbol{\epsilon}\|_\infty \sum_{t=r}^{\infty} \rho(\mathbf{C})^t = \|\boldsymbol{\epsilon}\|_\infty \frac{(1-\delta)^r}{\delta}.
|
| 482 |
+
\end{equation}
|
| 483 |
+
|
| 484 |
+
Setting $C_0 = \|\boldsymbol{\epsilon}\|_\infty / \delta$ completes the proof.
|
| 485 |
+
\end{proof}
|
| 486 |
+
|
| 487 |
+
\section{Proof of Proposition~\ref{prop:pg_constants} (Interaction Constants)}
|
| 488 |
+
\label{app:proof_constants}
|
| 489 |
+
|
| 490 |
+
\begin{proof}
|
| 491 |
+
Consider the CAVI update for user block $U_i$, component $k$:
|
| 492 |
+
\begin{equation}
|
| 493 |
+
a_{ik} = a_0 + \sum_{j' \in \Omega_i} x_{ij'} r_{ij'k}, \qquad
|
| 494 |
+
b_{ik} = b_0 + \sum_{j' \in \Omega_i} \frac{c_{j'k}}{d_{j'k}}.
|
| 495 |
+
\end{equation}
|
| 496 |
+
|
| 497 |
+
The dependence on item $j$'s parameters $(c_{jk}, d_{jk})$ enters through:
|
| 498 |
+
\begin{enumerate}[nosep]
|
| 499 |
+
\item The responsibility $r_{ijk}$, which depends on $\psi(c_{jk}) - \log d_{jk}$;
|
| 500 |
+
\item The rate update $b_{ik}$, which depends on $c_{jk}/d_{jk}$.
|
| 501 |
+
\end{enumerate}
|
| 502 |
+
|
| 503 |
+
For the shape parameter:
|
| 504 |
+
\begin{equation}
|
| 505 |
+
\frac{\partial a_{ik}}{\partial c_{j\ell}} = x_{ij} \frac{\partial r_{ijk}}{\partial c_{j\ell}}.
|
| 506 |
+
\end{equation}
|
| 507 |
+
The responsibility $r_{ijk}$ is a softmax of $\phi_{ijk} = \psi(a_{ik}) - \log b_{ik} + \psi(c_{jk}) - \log d_{jk}$.
|
| 508 |
+
Differentiating:
|
| 509 |
+
\begin{equation}
|
| 510 |
+
\frac{\partial r_{ijk}}{\partial c_{j\ell}} = r_{ijk}(\delta_{k\ell} - r_{ij\ell}) \psi'(c_{j\ell}).
|
| 511 |
+
\end{equation}
|
| 512 |
+
This gives $|\partial a_{ik}/\partial c_{j\ell}| \leq x_{ij} \cdot \psi'(c_{\min})/2$ using $|r_{ijk}(\delta_{k\ell} - r_{ij\ell})| \leq 1/2$.
|
| 513 |
+
|
| 514 |
+
For the rate parameter:
|
| 515 |
+
\begin{equation}
|
| 516 |
+
\frac{\partial b_{ik}}{\partial c_{jk}} = \frac{1}{d_{jk}}, \qquad
|
| 517 |
+
\frac{\partial b_{ik}}{\partial d_{jk}} = -\frac{c_{jk}}{d_{jk}^2}.
|
| 518 |
+
\end{equation}
|
| 519 |
+
|
| 520 |
+
Combining via the triangle inequality and summing over components, we obtain:
|
| 521 |
+
\begin{equation}
|
| 522 |
+
C_{U_i, V_j} \leq \underbrace{\left(\frac{1}{2}\psi'(c_{\min}) + \frac{1}{2d_{\min}}\right)}_{C_x} x_{ij} + \underbrace{\left(\frac{1}{d_{\min}} + \frac{c_{\max}}{d_{\min}^2}\right)}_{C_0}.
|
| 523 |
+
\end{equation}
|
| 524 |
+
|
| 525 |
+
The symmetric bound for $C_{V_j, U_i}$ follows by exchanging the roles of $(a, b)$ and $(c, d)$.
|
| 526 |
+
\end{proof}
|
| 527 |
+
|
| 528 |
+
\section{Local Unlearning Algorithm}
|
| 529 |
+
\label{app:algorithm}
|
| 530 |
+
|
| 531 |
+
\begin{algorithm}[h]
|
| 532 |
+
\caption{Local Radius-$R$ Unlearning for CAVI}
|
| 533 |
+
\label{alg:local}
|
| 534 |
+
\begin{algorithmic}[1]
|
| 535 |
+
\REQUIRE Full-data parameters $\blambda^\star$, deletion $z = (i, j, x_{ij})$, radius $R$
|
| 536 |
+
\STATE Compute seed set $S(z) = \{U_i, V_j\}$
|
| 537 |
+
\STATE Compute $R$-hop neighborhood $\cN_R(S) = \{u : d(u, S) \leq R\}$ via BFS
|
| 538 |
+
\STATE Filter edges: $E_{\mathrm{local}} = \{(i',j') \in E \setminus \{z\} : i' \in \cN_R \text{ or } j' \in \cN_R\}$
|
| 539 |
+
\STATE Initialize $\blambda \leftarrow \blambda^\star$
|
| 540 |
+
\REPEAT
|
| 541 |
+
\FOR{each block $u \in \cN_R(S)$}
|
| 542 |
+
\STATE $\blambda_u \leftarrow F_u^{\setminus z}(\blambda_{-u})$ \COMMENT{CAVI update using $E_{\mathrm{local}}$}
|
| 543 |
+
\ENDFOR
|
| 544 |
+
\UNTIL{convergence}
|
| 545 |
+
\RETURN $\blambda$
|
| 546 |
+
\end{algorithmic}
|
| 547 |
+
\end{algorithm}
|
| 548 |
+
|
| 549 |
+
The computational cost per CAVI iteration is $O(|E_{\mathrm{local}}| \cdot K)$ instead of $O(|E| \cdot K)$ for full retraining.
|
| 550 |
+
For bounded-degree graphs with maximum degree $d$, $|E_{\mathrm{local}}| = O(d^R)$, yielding a speedup of $O(|E| / d^R)$.
|
| 551 |
+
|
| 552 |
+
\section{Gaussian--Gaussian Interaction Constants}
|
| 553 |
+
\label{app:gaussian}
|
| 554 |
+
|
| 555 |
+
For the Gaussian--Gaussian model $x_{ij} \sim \cN(U_i^\top V_j, \sigma_x^2)$ with priors $U_{ik} \sim \cN(0, \sigma_U^2)$, $V_{jk} \sim \cN(0, \sigma_V^2)$, the mean-field CAVI updates for the variational mean are:
|
| 556 |
+
\begin{equation}
|
| 557 |
+
m^U_{ik} = s^U_{ik} \cdot \frac{1}{\sigma_x^2} \sum_{j' \in \Omega_i} m^V_{j'k} \left(x_{ij'} - \sum_{\ell \neq k} m^U_{i\ell} m^V_{j'\ell}\right),
|
| 558 |
+
\end{equation}
|
| 559 |
+
with precision $1/s^U_{ik} = 1/\sigma_U^2 + (1/\sigma_x^2) \sum_{j' \in \Omega_i} ((m^V_{j'k})^2 + s^V_{j'k})$.
|
| 560 |
+
|
| 561 |
+
Differentiating the update with respect to item $j$'s mean parameters:
|
| 562 |
+
\begin{equation}
|
| 563 |
+
\left\|\frac{\partial F_{U_i}}{\partial \blambda_{V_j}}\right\| \leq \frac{1}{\sigma_x^2} \|V_j\|_q = \frac{1}{\sigma_x^2} \sqrt{\sum_k ((m^V_{jk})^2 + s^V_{jk})},
|
| 564 |
+
\end{equation}
|
| 565 |
+
where the bound follows from the CAVI precision structure.
|
| 566 |
+
This gives the interaction proxy:
|
| 567 |
+
\begin{equation}
|
| 568 |
+
\chi_i^{GG} = \sum_{j' \in \Omega_i} \frac{1}{\sigma_x^2} \E_q[\|V_{j'}\|^2].
|
| 569 |
+
\end{equation}
|
| 570 |
+
|
| 571 |
+
\section{Additional Experimental Details}
|
| 572 |
+
\label{app:exp_details}
|
| 573 |
+
|
| 574 |
+
\paragraph{Synthetic data generation.}
|
| 575 |
+
For Gamma--Poisson data, we generate $U_{ik} \sim \Gam(a_0, b_0)$, $V_{jk} \sim \Gam(c_0, d_0)$, and $x_{ij} \sim \Poi(\rho_x U_i^\top V_j)$ for each observed edge.
|
| 576 |
+
Zero counts are removed (positive counts only).
|
| 577 |
+
Graph families: bounded-degree (Poisson degree truncated at $2d_{\max}$), Erd\H{o}s--R\'enyi ($p = d_{\mathrm{avg}}/M$), power-law (degree $\propto k^{-\alpha+1}$ with $\alpha = 2.5$).
|
| 578 |
+
|
| 579 |
+
\paragraph{Deletion sampling.}
|
| 580 |
+
We sample 25\% each of: random edges, high-count edges (top-quartile by $x_{ij}$), hub-adjacent edges (top-quartile by $\max(\deg(i), \deg(j))$), and low-degree edges (bottom-quartile by $\min(\deg(i), \deg(j))$).
|
| 581 |
+
|
| 582 |
+
\paragraph{Convergence.}
|
| 583 |
+
CAVI: relative parameter change $< 10^{-5}$, max 200--300 iterations.
|
| 584 |
+
GGM MAP: Adam with $\beta_1 = 0.9$, $\beta_2 = 0.999$, lr $= 0.05$, gradient clipping at 10, max 2000 iterations.
|
| 585 |
+
|
| 586 |
+
\paragraph{Bootstrap confidence intervals.}
|
| 587 |
+
All reported confidence intervals are 95\% bootstrap CIs with 1000 resamples.
|
| 588 |
+
|
| 589 |
+
\paragraph{Datasets.}
|
| 590 |
+
\begin{itemize}[nosep]
|
| 591 |
+
\item Last.fm: \texttt{matthewfranglen/lastfm-1k} on Hugging Face. 2M rows randomly sampled, aggregated to user--artist counts, capped at 50, filtered to $\geq 5$ degree.
|
| 592 |
+
\item MovieLens: \texttt{ashraq/movielens\_ratings} on Hugging Face. Rating count: $x_{ij} = \lceil \mathrm{rating} \rceil$. Binary: $x_{ij} = 1$.
|
| 593 |
+
\end{itemize}
|
| 594 |
+
|
| 595 |
+
\section{Full Sanity Check Results}
|
| 596 |
+
\label{app:sanity}
|
| 597 |
+
|
| 598 |
+
\begin{table}[h]
|
| 599 |
+
\centering
|
| 600 |
+
\caption{Sanity check pass rates across 270 verified deletions.}
|
| 601 |
+
\small
|
| 602 |
+
\begin{tabular}{lcc}
|
| 603 |
+
\toprule
|
| 604 |
+
Check & PG (198) & All (270) \\
|
| 605 |
+
\midrule
|
| 606 |
+
Parameters positive & 100\% & -- \\
|
| 607 |
+
No NaN & 100\% & -- \\
|
| 608 |
+
Responsibilities $\sum_k = 1$ & 100\% & -- \\
|
| 609 |
+
ELBO finite & 100\% & -- \\
|
| 610 |
+
Exact $\neq$ full & 88\% & 90\% \\
|
| 611 |
+
Error $\downarrow$ with $R$ & 90\% & 90\% \\
|
| 612 |
+
Warm-start $\approx$ large-$R$ & 91\% & 91\% \\
|
| 613 |
+
\bottomrule
|
| 614 |
+
\end{tabular}
|
| 615 |
+
\end{table}
|
| 616 |
+
|
| 617 |
+
The 10\% of error-monotonicity violations occur predominantly in the Gaussian--Gamma MAP model (non-convex optimization) and in high-coupling regimes where the Dobrushin condition is marginal.
|
| 618 |
+
|
| 619 |
+
\end{document}
|