To Retain or to Adapt? Generalizing Continual Learning β part 10/15
Cleaned-text code points [141123, 154774). Bibliography omitted here; source.md includes it.
β’ πΌ β‘ [ | Tr β ( Ξ£ ~ 0 JTL , k ) | ] \mathbb{E}[|\text{Tr}(\tilde{\Sigma}_{0}^{\text{JTL},k})|] : Similarly, by Lemma 28 and Lemma 31 , the covariance initialization error is dominated by the covariance drift Ξ β ( Ο β² β ( k ) ) \Theta(\psi^{\prime}(k)) .
Substituting these scalings back into the inequality for sufficiently large k k : Ο β N β
Ξ β‘ ( Ο β‘ ( k ) ) < C ITL k β ( Ξ β‘ ( Ο β‘ ( k ) ) + Ξ β‘ ( 1 ) ) β C JTL k β ( Ξ β‘ ( Ο β‘ ( k ) k 2 ) + Ξ β‘ ( Ο β² β ( k ) ) ) . \tau N\cdot\Theta(\psi(k))<C^{k}_{\text{ITL}}\left(\Theta(\psi(k))+\Theta(1)\right)-C^{k}_{\text{JTL}}\left(\Theta\left(\frac{\psi(k)}{k^{2}}\right)+\Theta(\psi^{\prime}(k))\right). The JTL transient term (the negative contribution) becomes asymptotically negligible compared to the ITL transient term (the positive contribution) because Ο β‘ ( k ) k 2 βͺ Ο β‘ ( k ) \frac{\psi(k)}{k^{2}}\ll\psi(k) and Ο β² β ( k ) βͺ Ο β‘ ( k ) \psi^{\prime}(k)\ll\psi(k) for any polynomial growth.
Therefore, the asymptotic condition simplifies to a direct competition between the ITL transient energy and the JTL instability: Ο β N β
πΌ β‘ [ β k ] < C ITL k β
πΌ β‘ [ β ΞΌ ~ 0 ITL β 2 ] . \tau N\cdot\mathbb{E}[\mathcal{I}^{k}]<C^{k}_{\text{ITL}}\cdot\mathbb{E}[\|\tilde{\mu}_{0}^{\text{ITL}}\|^{2}]. Rearranging for N N , positive expected transfer occurs only if the task duration is sufficiently short relative to the instability: N < C ITL k Ο β πΌ β‘ [ β ΞΌ ~ 0 ITL β 2 ] πΌ β‘ [ β k ] . N<\frac{C^{k}_{\text{ITL}}}{\tau}\frac{\mathbb{E}[\|\tilde{\mu}_{0}^{\text{ITL}}\|^{2}]}{\mathbb{E}[\mathcal{I}^{k}]}. This concludes the proof: expected transfer is positive strictly in the "transient regime", where the benefit of initializing near the task centroid (avoiding re-learning) outweighs the cost of the asymptotic bias and task conflict. β
#### A.6 (Non-asymptotic) Continuous Drift Analysis in the Quadratic Model
We now instantiate the general Transfer Efficiency framework for the specific case of drifting linear tasks. This allows us to derive precise, non-asymptotic conditions for the stability-plasticity phase transition observed in our experiments.
##### A.6.1 Setup and Definitions
We assume the environment consists of a sequence of tasks k = 1 , β¦ , K k=1,\dots,K governed by a random walk process. The data inputs x β β d x\in\mathbb{R}^{d} are drawn from a fixed distribution with covariance matrix Ξ£ x β» 0 \Sigma_{x}\succ 0 . The target for task k k is given by y = ( ΞΈ β k ) β€ β x + ΞΎ y=(\theta_{\star}^{k})^{\top}x+\xi , with label noise ΞΎ βΌ π© β‘ ( 0 , Ο ΞΎ 2 ) \xi\sim\mathcal{N}(0,\sigma_{\xi}^{2}) . The expected loss function for task k k (over the data distribution) is: β k β ( ΞΈ ) = 1 2 β β ΞΈ β ΞΈ β k β Ξ£ x 2 + 1 2 β Ο ΞΎ 2 , \mathcal{L}^{k}(\theta)=\frac{1}{2}\|\theta-\theta_{\star}^{k}\|_{\Sigma_{x}}^{2}+\frac{1}{2}\sigma_{\xi}^{2}, (77) using the Mahalanobis-style norm β v β M 2 := v β€ β M β v \|v\|_{M}^{2}:=v^{\top}Mv introduced in the main text ( Section 4 ). This expression is the instantiation of the quadratic loss of Section 4 with H j = Ξ£ x H_{j}=\Sigma_{x} (the same data Hessian shared across tasks) and the irreducible loss c j = 1 2 β Ο ΞΎ 2 c_{j}=\tfrac{1}{2}\sigma_{\xi}^{2} . The optimal parameters evolve according to a Wiener process across tasks: ΞΈ β k = ΞΈ β k β 1 + Ξ k , Ξ k βΌ π© β‘ ( 0 , Ξ£ Ξ ) , \theta_{\star}^{k}=\theta_{\star}^{k-1}+\Delta_{k},\quad\Delta_{k}\sim\mathcal{N}(0,\Sigma_{\Delta}), (78) where Ξ£ Ξ \Sigma_{\Delta} is the drift covariance matrix. Both the Independent-Task Learning (ITL) and Joint-Task Learning (JTL) agents optimize these objectives using SGD with a constant step-size Ξ· \eta . For simplicity, we assume all tasks have equal duration N N .
As we will demonstrate in the derivations below, this type of non-stationarity exhibits a Diffusive growth rate ( Ο β‘ ( k ) = k \psi(k)=k ), consistent with the regimes defined in Table 4 . Unless otherwise noted, expectations πΌ β‘ [ β
] \mathbb{E}[\cdot] in this section are taken over the random walk process defining the task sequence.
##### A.6.2 Derivation of Expected Instability
The JTL agent optimizes the uniform mixture of all observed tasks, converging to the cumulative centroid ΞΈ β JTL , k = 1 k β β j = 1 k ΞΈ β j \theta_{\star}^{\text{JTL},k}=\frac{1}{k}\sum_{j=1}^{k}\theta_{\star}^{j} . We apply the Exact Instability Decomposition from Theorem 24 to task k k . Since the Hessian H k = Ξ£ x H_{k}=\Sigma_{x} is constant across tasks, the instability for a given realization decomposes into: β k β 1 2 β β ΞΈ β JTL , k β ΞΈ β k β Ξ£ x 2 + Ξ· 4 β k β B β β j = 1 k β ΞΈ β JTL , k β ΞΈ β j β Ξ£ x 2 2 . \mathcal{I}^{k}\approx\frac{1}{2}\|\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{k}\|_{\Sigma_{x}}^{2}+\frac{\eta}{4kB}\sum_{j=1}^{k}\|\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{j}\|_{\Sigma_{x}^{2}}^{2}. (79)
We first express the lag vector v k := ΞΈ β JTL , k β ΞΈ β k v_{k}:=\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{k} in terms of the sequence of drift increments Ξ m \Delta_{m} . Using the backward recursion ΞΈ β j = ΞΈ β k β β m = j + 1 k Ξ m \theta_{\star}^{j}=\theta_{\star}^{k}-\sum_{m=j+1}^{k}\Delta_{m} , we substitute into the centroid definition: v k = ( 1 k β j = 1 k ΞΈ β j ) β ΞΈ β k = 1 k β j = 1 k ( ΞΈ β k β β m = j + 1 k Ξ m ) β ΞΈ β k = β 1 k β j = 1 k β m = j + 1 k Ξ m . v_{k}=\left(\frac{1}{k}\sum_{j=1}^{k}\theta_{\star}^{j}\right)-\theta_{\star}^{k}=\frac{1}{k}\sum_{j=1}^{k}\left(\theta_{\star}^{k}-\sum_{m=j+1}^{k}\Delta_{m}\right)-\theta_{\star}^{k}=-\frac{1}{k}\sum_{j=1}^{k}\sum_{m=j+1}^{k}\Delta_{m}. (80) Rearranging the double summation to group terms by the drift increment Ξ m \Delta_{m} (noting that a specific Ξ m \Delta_{m} appears whenever j < m j<m , meaning it appears m β 1 m-1 times): v k = β 1 k β m = 2 k ( m β 1 ) Ξ m . v_{k}=-\frac{1}{k}\sum_{m=2}^{k}(m-1)\Delta_{m}. (81)
The second term in the instability equation involves the sum of squared deviations from the centroid (the task scatter). This can be rewritten using pairwise squared differences: β j = 1 k ( ΞΈ β JTL , k β ΞΈ β j ) β ( ΞΈ β JTL , k β ΞΈ β j ) β€ = 1 2 β k β β i = 1 k β j = 1 k ( ΞΈ β i β ΞΈ β j ) β ( ΞΈ β i β ΞΈ β j ) β€ . \sum_{j=1}^{k}(\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{j})(\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{j})^{\top}=\frac{1}{2k}\sum_{i=1}^{k}\sum_{j=1}^{k}(\theta_{\star}^{i}-\theta_{\star}^{j})(\theta_{\star}^{i}-\theta_{\star}^{j})^{\top}. (82) The difference between any two optima is simply the sum of the intermediate drift steps: ΞΈ β i β ΞΈ β j = β m = j + 1 i Ξ m \theta_{\star}^{i}-\theta_{\star}^{j}=\sum_{m=j+1}^{i}\Delta_{m} (for i > j i>j ).
We now compute the expected instability πΌ β‘ [ β k ] \mathbb{E}[\mathcal{I}^{k}] over the distribution of random walks. The covariance of the lag vector v k v_{k} is the weighted sum of the increment covariances: πΌ β‘ [ v k β v k β€ ] = 1 k 2 β β m = 2 k ( m β 1 ) 2 β Ξ£ Ξ = 1 k 2 β ( ( k β 1 ) β k β ( 2 β k β 1 ) 6 ) β Ξ£ Ξ . \mathbb{E}[v_{k}v_{k}^{\top}]=\frac{1}{k^{2}}\sum_{m=2}^{k}(m-1)^{2}\Sigma_{\Delta}=\frac{1}{k^{2}}\left(\frac{(k-1)k(2k-1)}{6}\right)\Sigma_{\Delta}. (83) Using the continuous approximation β p = 1 k p 2 β k 3 / 3 \sum_{p=1}^{k}p^{2}\approx k^{3}/3 for large k k : πΌ β‘ [ 1 2 β β v k β Ξ£ x 2 ] β 1 2 β Tr β ( Ξ£ x β
k 3 β Ξ£ Ξ ) = k 6 β Tr β ( Ξ£ x β Ξ£ Ξ ) . \mathbb{E}\left[\frac{1}{2}\|v_{k}\|_{\Sigma_{x}}^{2}\right]\approx\frac{1}{2}\text{Tr}\left(\Sigma_{x}\cdot\frac{k}{3}\Sigma_{\Delta}\right)=\frac{k}{6}\text{Tr}(\Sigma_{x}\Sigma_{\Delta}). (84)
For the task scatter, since πΌ β‘ [ ( ΞΈ β i β ΞΈ β j ) β ( ΞΈ β i β ΞΈ β j ) β€ ] = | i β j | β Ξ£ Ξ \mathbb{E}[(\theta_{\star}^{i}-\theta_{\star}^{j})(\theta_{\star}^{i}-\theta_{\star}^{j})^{\top}]=|i-j|\Sigma_{\Delta} , the expected sum of pairwise distances is: πΌ β‘ [ Scatter ] = 1 2 β k β β i , j | i β j | β Ξ£ Ξ β 1 2 β k β ( k 3 3 ) β Ξ£ Ξ = k 2 6 β Ξ£ Ξ . \mathbb{E}[\text{Scatter}]=\frac{1}{2k}\sum_{i,j}|i-j|\Sigma_{\Delta}\approx\frac{1}{2k}\left(\frac{k^{3}}{3}\right)\Sigma_{\Delta}=\frac{k^{2}}{6}\Sigma_{\Delta}. (85) Substituting this into the variance trace term: πΌ β‘ [ Ξ· 4 β k β B β β j = 1 k β ΞΈ β JTL , k β ΞΈ β j β Ξ£ x 2 2 ] = Ξ· 4 β k β B β Tr β ( Ξ£ x 2 β
k 2 6 β Ξ£ Ξ ) = Ξ· β k 24 β B β Tr β ( Ξ£ x 2 β Ξ£ Ξ ) . \mathbb{E}\left[\frac{\eta}{4kB}\sum_{j=1}^{k}\|\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{j}\|_{\Sigma_{x}^{2}}^{2}\right]=\frac{\eta}{4kB}\text{Tr}\left(\Sigma_{x}^{2}\cdot\frac{k^{2}}{6}\Sigma_{\Delta}\right)=\frac{\eta k}{24B}\text{Tr}(\Sigma_{x}^{2}\Sigma_{\Delta}). (86)
Combining the expected bias and variance components, the final expected instability is: πΌ β‘ [ β k ] β k 6 β Tr β ( Ξ£ x β Ξ£ Ξ ) + Ξ· β k 24 β B β Tr β ( Ξ£ x 2 β Ξ£ Ξ ) . \mathbb{E}[\mathcal{I}^{k}]\approx\frac{k}{6}\text{Tr}(\Sigma_{x}\Sigma_{\Delta})+\frac{\eta k}{24B}\text{Tr}(\Sigma_{x}^{2}\Sigma_{\Delta}). (87)
##### A.6.3 Derivation of Expected Transient Error
The Transient Error Ξ΄ k \delta^{k} is proportional to the energy of the initial parameter error relative to the current taskβs stationary distribution. For a specific realization, this is approximated by: Ξ΄ k β 1 N β Ο β Tr β ( ΞΎ N β ( Ξ£ x ) β ( ΞΌ ~ 0 β ΞΌ ~ 0 β€ + Ξ£ ~ 0 ) ) . \delta^{k}\approx\frac{1}{N\tau}\text{Tr}\left(\xi_{N}(\Sigma_{x})\left(\tilde{\mu}_{0}\tilde{\mu}_{0}^{\top}+\tilde{\Sigma}_{0}\right)\right). (88)
###### ITL Agent.
The ITL agent resets to a fixed prior ΞΈ 0 βΌ Q 0 \theta_{0}\sim Q_{0} with covariance Ξ£ 0 ITL \Sigma_{0}^{\text{ITL}} . The initial error vector is ΞΌ ~ 0 ITL = ΞΈ 0 β ΞΈ β k \tilde{\mu}_{0}^{\text{ITL}}=\theta_{0}-\theta_{\star}^{k} . The expected energy of this error depends entirely on how far the random walk has drifted from the origin: πΌ β‘ [ β ΞΈ 0 β ΞΈ β k β 2 ] = πΌ β‘ [ β ΞΈ 0 β 2 ] + Tr β ( β j = 1 k πΌ β‘ [ Ξ j β Ξ j β€ ] ) = Tr β ( Ξ£ 0 ITL ) + k β Tr β ( Ξ£ Ξ ) . \mathbb{E}[\|\theta_{0}-\theta_{\star}^{k}\|^{2}]=\mathbb{E}[\|\theta_{0}\|^{2}]+\text{Tr}\left(\sum_{j=1}^{k}\mathbb{E}[\Delta_{j}\Delta_{j}^{\top}]\right)=\text{Tr}\left(\Sigma_{0}^{\text{ITL}}\right)+k\text{Tr}(\Sigma_{\Delta}). (89) Let Ξ³ min := Ξ» min β ( ΞΎ N β ( Ξ£ x ) ) \gamma_{\min}:=\lambda_{\min}(\xi_{N}(\Sigma_{x})) . Because Ξ£ ~ 0 ITL = Ξ£ 0 ITL β Ξ£ β ITL \tilde{\Sigma}_{0}^{\text{ITL}}=\Sigma_{0}^{\text{ITL}}-\Sigma_{\infty}^{\text{ITL}} , the expected ITL Transient Error is bounded by: πΌ β‘ [ Ξ΄ ITL k ] \displaystyle\mathbb{E}[\delta^{k}_{\text{ITL}}] β₯ Ξ³ min Ο β N β ( πΌ β‘ [ β ΞΌ ~ 0 ITL β 2 ] + Tr β ( Ξ£ ~ 0 ITL ) ) \displaystyle\geq\frac{\gamma_{\min}}{\tau N}\left(\mathbb{E}[\|\tilde{\mu}_{0}^{\text{ITL}}\|^{2}]+\text{Tr}(\tilde{\Sigma}_{0}^{\text{ITL}})\right) = Ξ³ min Ο β N β ( k β Tr β ( Ξ£ Ξ ) + 2 β Tr β ( Ξ£ 0 ITL ) β Tr β ( Ξ£ β ITL ) ) . \displaystyle=\frac{\gamma_{\min}}{\tau N}\left(k\text{Tr}(\Sigma_{\Delta})+2\text{Tr}\left(\Sigma_{0}^{\text{ITL}}\right)-\text{Tr}({\Sigma}^{\text{ITL}}_{\infty})\right). (90) By Lemma 20 , Tr β ( Ξ£ β ITL ) = Ξ· β Ο ΞΎ 2 2 β B \text{Tr}({\Sigma}^{\text{ITL}}_{\infty})=\frac{\eta\sigma_{\xi}^{2}}{2B} . The trace of the prior Tr β ( Ξ£ 0 ITL ) \text{Tr}\left(\Sigma_{0}^{\text{ITL}}\right) is a constant, which we denote by Ο 0 2 \sigma_{0}^{2} .
###### JTL Agent.
The JTL agent initializes at the previous taskβs converged centroid ΞΈ β JTL , k β 1 \theta_{\star}^{\text{JTL},k-1} . The expected initialization error decomposes into the geometric drift and the residual contraction: ΞΌ ~ 0 JTL , k = Ξ β ΞΌ k + r ΞΌ k β 1 . \tilde{\mu}_{0}^{\text{JTL},k}=\Delta\mu_{k}+r_{\mu}^{k-1}. First, we relate the geometric drift Ξ β ΞΌ k \Delta\mu_{k} to the lag vector v k v_{k} . Recall ΞΈ β JTL , k = 1 k β ΞΈ β k + k β 1 k β ΞΈ β JTL , k β 1 \theta_{\star}^{\text{JTL},k}=\frac{1}{k}\theta_{\star}^{k}+\frac{k-1}{k}\theta_{\star}^{\text{JTL},k-1} . Rearranging for the increment: Ξ β ΞΌ k = ΞΈ β JTL , k β ΞΈ β JTL , k β 1 = ΞΈ β JTL , k β k k β 1 β ( ΞΈ β JTL , k β 1 k β ΞΈ β k ) = 1 k β 1 β ( ΞΈ β k β ΞΈ β JTL , k ) = β 1 k β 1 β v k . \Delta\mu_{k}=\theta_{\star}^{\text{JTL},k}-\theta_{\star}^{\text{JTL},k-1}=\theta_{\star}^{\text{JTL},k}-\frac{k}{k-1}\left(\theta_{\star}^{\text{JTL},k}-\frac{1}{k}\theta_{\star}^{k}\right)=\frac{1}{k-1}(\theta_{\star}^{k}-\theta_{\star}^{\text{JTL},k})=-\frac{1}{k-1}v_{k}. The expected squared norm of the geometric drift is therefore: πΌ β‘ [ β Ξ β ΞΌ k β 2 ] = 1 ( k β 1 ) 2 β Tr β ( πΌ β‘ [ v k β v k β€ ] ) = 2 β k β 1 6 β k β ( k β 1 ) β Tr β ( Ξ£ Ξ ) . \mathbb{E}[\|\Delta\mu_{k}\|^{2}]=\frac{1}{(k-1)^{2}}\text{Tr}(\mathbb{E}[v_{k}v_{k}^{\top}])=\frac{2k-1}{6k(k-1)}\text{Tr}(\Sigma_{\Delta}). (91) Let d k := πΌ β‘ [ β Ξ β ΞΌ k β 2 ] d_{k}:=\sqrt{\mathbb{E}[\|\Delta\mu_{k}\|^{2}]} be the RMS drift intensity. For k β₯ 2 k\geq 2 , define the RMS initialization error e k := πΌ β‘ [ β ΞΌ ~ 0 JTL , k β 2 ] e_{k}:=\sqrt{\mathbb{E}[\|\tilde{\mu}_{0}^{\text{JTL},k}\|^{2}]} . By Minkowskiβs inequality, it satisfies the stable recursion e k β€ d k + Ο ΞΌ β e k β 1 e_{k}\leq d_{k}+\rho_{\mu}e_{k-1} . Because the drift d k d_{k} changes slowly compared to the exponential decay rate Ο ΞΌ \rho_{\mu} (the quasi-static approximation), the error tracks the input scaled by the effective time constant 1 1 β Ο ΞΌ \frac{1}{1-\rho_{\mu}} . For k β« 1 k\gg 1 , we approximate d k β 1 3 β k β Tr β ( Ξ£ Ξ ) d_{k}\approx\sqrt{\frac{1}{3k}\text{Tr}(\Sigma_{\Delta})} . This yields the explicit expected transient energy curve: πΌ β‘ [ β ΞΌ ~ 0 JTL , k β 2 ] β ( 1 1 β Ο ΞΌ ) 2 β Tr β ( Ξ£ Ξ ) 3 β k . \mathbb{E}[\|\tilde{\mu}_{0}^{\text{JTL},k}\|^{2}]\approx\left(\frac{1}{1-\rho_{\mu}}\right)^{2}\frac{\text{Tr}(\Sigma_{\Delta})}{3k}. (92) This confirms that the JTL expected initialization error vanishes as 1 / k 1/k , but its magnitude is inflated by the conditioning of the optimization problem via the ( 1 β Ο ΞΌ ) β 2 (1-\rho_{\mu})^{-2} factor.