mirror of
https://github.com/hanzoai/papers.git
synced 2026-11-03 02:27:30 +00:00
38 lines
1.5 KiB
TeX
38 lines
1.5 KiB
TeX
\section{Proof of Theorem~\ref{thm:rate}
|
|
(Full)}
|
|
\label{app:proof-rate}
|
|
|
|
\begin{proof}
|
|
Fix token position $t$ and let $v^*$ denote the Bayes-optimal token. For each
|
|
prior $m$, define the log-odds contribution:
|
|
\begin{equation}
|
|
X_m = \eta_m \left[\log \phi_m(v^* \mid x, y_{<t}) - \log \phi_m(v \mid x, y_{<t})\right]
|
|
\end{equation}
|
|
for any $v \neq v^*$. By construction, $X_m \geq 0$ when $\phi_m$ correctly
|
|
assigns higher probability to $v^*$ (which happens with probability $q_m$) and
|
|
$|X_m| \leq \eta_m \cdot \max_v |\log \phi_m(v)| \leq C$ for some constant $C$
|
|
determined by the maximum magnitude of the expert factors.
|
|
|
|
The total log-odds advantage for $v^*$ over $v$ after $M$ priors is:
|
|
\begin{equation}
|
|
L_M(v^*, v) = \ell_{t,v^*} - \ell_{t,v} + \sum_{m=1}^M X_m.
|
|
\end{equation}
|
|
|
|
By assumption, $\mathbb{E}[X_m] \geq 2\delta^2$ (from $q_m > 0.5 + \delta$
|
|
and the log-odds formula). By Hoeffding's inequality:
|
|
\begin{equation}
|
|
P\left(\sum_{m=1}^M X_m < M\delta^2\right)
|
|
\leq \exp\left(-\frac{M\delta^2}{2C^2}\right)
|
|
\leq \exp\left(-\frac{M\delta^2}{2}\right)
|
|
\end{equation}
|
|
for $C = 1$ (which holds when expert factors are bounded by 1 in absolute
|
|
value). Taking a union bound over all $V - 1$ incorrect tokens and noting
|
|
that the KL divergence is zero when $\pi_{\text{ASO}}(v^*) = 1$:
|
|
\begin{equation}
|
|
P\left(\KL(\pi^* \| \pi_{\text{ASO}}^{(M)}) > \varepsilon\right)
|
|
\leq (V-1) \exp\left(-\frac{M\delta^2}{2}\right)
|
|
< 2V \exp\left(-\frac{M\delta^2}{2}\right).
|
|
\end{equation}
|
|
\end{proof}
|
|
|