4.3.12 · HinglishPretraining & Fine-Tuning LLMs

Catastrophic forgetting

1,766 words8 min readRead in English

4.3.12 · AI-ML › Pretraining & Fine-Tuning LLMs


YEH KYUN hota hai? (First principles)

Ek neural network jo kuch bhi jaanta hai woh sab ek shared set of weights mein store hota hai. Har task ke liye koi alag "memory slot" nahi hota — knowledge distributed aur overlapping hoti hai.

Jab tum ek naye dataset par fine-tune karte ho, tum ek aisi loss minimize karte ho jo sirf naye data ka zikr karti hai:

Gradient descent yeh steps leta hai:

Is update mein purane task ki loss ka koi reference nahi hota. Toh optimizer khushi se ko us region se bahar le jaata hai jo purane task ke liye achha tha, jab tak naya task improve hota rahe. Purani skill collateral damage ban jaati hai.


YEH DIKHAI KAISE deta hai?

Do loss landscapes imagine karo ek hi weight vector par:

  • — ek valley jiska bottom "task A mein achha" region hai.
  • — task B ke liye ek alag valley.

Inke minima zyada tar alag-alag jagah hote hain. Pure fine-tuning ko naye valley mein slide kar deti hai, purani valley se bahar nikalte hue.

Figure — Catastrophic forgetting

FIX KAISE derive karte hain? (EWC scratch se)

Goal: naye task par train karo bina purane task ke achhe region se bahar nikalte hue. Hum chahte hain ek aisa jo naye par achha ho aur purane optimum ke paas rahe, lekin sirf un directions mein "stiff" ho jo purane task ke liye matter karte the.

Step 1 — Bayesian view. Purana task seekhne ke baad hamare paas posterior hai. Jab naya data aata hai, Bayes kehta hai:

Yeh step kyun? Purane data ka influence poori tarah posterior mein capture ho jaata hai — purana data paas rakhne ki zaroorat nahi, bas yaad rakho ki usne ke baare mein kya bataya.

Step 2 — Purane posterior ko Gaussian se approximate karo (Laplace approximation) par centered, jisme precision Fisher information se milti hai:

Yeh step kyun? Kisi bhi minimum ke paas smooth loss quadratic dikhti hai; curvature (2nd derivative) hume batati hai ki har weight kitna important tha. bada ⇒ weight bahut mattered ⇒ use mat hilao.

Step 3 — Ek trainable loss mein combine karo (Elastic Weight Consolidation):


Doosri mitigation strategies (80/20)

Woh 20% jo tumhe zaroor pata hona chahiye:

Method Core idea WHY it works
Rehearsal / Replay Training mein purane data ka thoda sa hissa (ya generated pseudo-data) mix karo Purani loss ab objective mein hai, toh gradients uski raksha karte hain
EWC / regularization Important weights ko move karne par penalty do ko key directions mein purane optimum ke paas rehne par restrict karta hai
Parameter isolation (LoRA/adapters) Base weights freeze karo; chote naye trainable modules add karo Purane weights literally change nahi ho sakte ⇒ forget nahi ho sakte
Lower LR + fewer epochs se chote steps lo Purani valley se kam drift

Steel-manned mistakes


Recall Feynman: 12-saal ke bacche ko samjhao

Socho tumhara brain mitti ka ek lump hai aur jo bhi skill tum seekhte ho woh uspe ek shape dabata hai. Agar tum cycle chalana seekhte ho, phir usi jagah swimming seekhne ke liye bahut zor se dabate ho, toh bike-wali shape flat ho sakti hai aur tum use bhool sakte ho! Catastrophic forgetting exactly yahi hai — computer ek hi "clay weights" ka lump sab cheez ke liye use karta hai, toh naye kaam par zor se dabane se purana kaam erase ho sakta hai. Ise fix karne ke liye ya toh hum (a) purani cheez thodi practice karte rehte hain (rehearsal), ya (b) mitti ke un hisson par chote "mujhe rakho!" springs lagate hain jo important purani shapes hold karte hain (EWC), ya (c) naya skill ek nayi mitti ke tedon mein seekhte hain jo side mein lagayi ho (adapters/LoRA).


Connections

  • Fine-Tuning LLMs — jahan forgetting sabse zyada kaatti hai.
  • LoRA and Adapters — parameter-isolation fix.
  • Fisher Information — EWC mein "importance" .
  • Laplace Approximation — Gaussian posterior kaise aata hai.
  • Continual Learning — broader problem setting.
  • Multi-task Learning — "sab kuch jointly train karo" ideal jise EWC/rehearsal approximate karte hain.
  • Learning Rate & Optimization — chota LR drift kam karta hai.

Flashcards

Catastrophic forgetting kya hai?
Pehle seekhi hui knowledge ka abrupt loss jab network ko naye task par train kiya jaata hai, kyunki gradient updates shared weights ko overwrite kar dete hain jo purane task ko encode karte the.
Naye data par fine-tuning se forgetting kyun hoti hai?
Nayi loss mein purane task ka koi term nahi hota, toh gradient descent freely ko purane task ke achhe region se bahar le jaata hai.
EWC loss likho.
EWC mein Fisher information kya represent karta hai?
Purane task ke liye weight ki importance/curvature; bada ⇒ woh weight matter karta tha ⇒ use hilane par penalty do.
EWC saari learning kyun block nahi karta?
Purane task ke liye unimportant weights ka hota hai, toh woh change karne ke liye free hain — learning spare capacity mein hoti hai.
Teen families of mitigation batao.
Rehearsal/replay, regularization (EWC), aur parameter isolation (LoRA/adapters).
LoRA by design forgetting kyun avoid karta hai?
Yeh base weights freeze kar deta hai aur sirf chote added modules train karta hai, toh original knowledge weights literally overwrite nahi ho sakti.
Steel-man: zyada epochs forgetting kyun badhaate hain?
Naye task par extra epochs ko purane optimum se aur door push karte hain, purane task ki valley se drift badh jaati hai.
EWC quadratic penalty mathematically kahan se aati hai?
Purane-task posterior ka Laplace (Gaussian) approximation ke around, jisme curvature milti hai.

Concept Map

encode

encode

minimizes

gradient descent

ignores

overwrites

abrupt loss

fix via

derived from

Laplace approx

weighted by

large F stiffens

Shared weights theta

Old task knowledge

New task knowledge

Fine-tune on new data

Loss on new data only

Update theta

Old task loss

Catastrophic forgetting

Elastic Weight Consolidation

Bayesian posterior

Gaussian around theta star

Fisher information F_i