4.4.2 · HinglishAlignment, Prompting & RAG

Reward modeling

1,676 words8 min readRead in English

4.4.2 · AI-ML › Alignment, Prompting & RAG


WHY reward modeling exist karta hai?


WHAT hota hai ek reward model?


HOW karte hain hum "A beats B" ko ek loss mein convert? (Scratch se derivation)

Humein preferences ka ek probabilistic model chahiye. First principles se shuru karte hain.

Step 1 — Maano har response ka ek hidden "quality" score hota hai. Reward model assign kare aur . Kyun? Humein do scalars ko ek probability mein convert karna hai ki choose hoga.

Step 2 — Bradley–Terry model. Maano ki ek human ke prefer karne ki probability sirf score difference par depend karti hai, monotonically. Standard choice: probability ke saath ek logistic (sigmoid) ke through badhti hai.

Sigmoid kyun? ek positive "strength" hai. Winner ki strength ka share hai. Upar aur neeche se divide karne par milta hai . Toh sirf difference matter karta hai, absolute scale nahi.

Step 3 — Likelihood maximize karo = negative log-likelihood minimize karo. Ek example ke liye observed human choice ki likelihood hai. lo:

Step 4 — Check karo ki gradient sensible hai. Maano . use karke: Kyun sensible hai? Agar model already confident hai ( bada positive), — tiny gradient, "kuch fix nahi karna". Agar galat hai (), bada hai — bada correction. Self-correcting.

Figure — Reward modeling

Worked examples


Training ke baad: reward model use kaise hota hai


Common mistakes (Steel-manned)


Flashcards

RLHF mein reward model kya hota hai?
Ek learned scalar function jo human preferences par train hota hai taaki score kare ki response prompt ke liye kitna achha hai; usually base LM with a scalar output head.
"Be helpful" ke liye reward haath se kyun nahi likh sakte?
Yeh property pahchaanno mein aasaan hai lekin formulaically specify karna impossible hai; humans outputs compare kar sakte hain, toh hum reward comparisons se seekhte hain.
Bradley–Terry preference probability batao.
.
Pairwise reward-model loss likho.
.
Sirf reward difference kyun matter karta hai, absolute value kyun nahi?
Sigmoid sirf par depend karta hai; saare rewards mein constant add karne se loss unchanged rehta hai (scale/shift invariant).
ke w.r.t. gradient kya hai aur yeh self-correcting kyun hai?
; bada jab model galat ho (), near zero jab confidently sahi ho.
K responses ki ranking training data mein kaise convert hoti hai?
Saare pairs banao aur unke pairwise losses average karo.
Reward hacking / over-optimization kya hai?
Policy un regions exploit karti hai jahan RM confidently galat hai, measured reward badhta hai jabki true preference girti hai (Goodhart).
RM ke against optimize karte waqt KL penalty kyun include karte hain?
Policy ko reference distribution ke paas rakhne ke liye taaki RM in-distribution rahe aur exploit na ho sake.
Pairwise comparison loss ko ratings par MSE regression se prefer kyun karte hain?
Comparisons absolute human scores se zyaada reliable hain; logistic loss scale-free hai.

Recall Feynman: 12-saal ke bachche ko samjhao

Socho tum ek robot ko achhe birthday cards likhna sikha rahe ho, lekin "achha" define karna mushkil hai. Toh uski jagah tum usse do cards dikhate ho aur bas woh wala point karte ho jo tumhe zyaada pasand hai, baar baar. Tumhari saari pointing se robot ek chhota sa "achhaai meter" banata hai jo har card ko score deta hai. Koi farak nahi padhta ki numbers 5 aur 3 hain ya 105 aur 103 — jo matter karta hai woh yeh hai ki tumhara favourite zyaada number paye. Baad mein robot naaye cards likhta hai meter ko upar le jaane ki koshish mein. Lekin dhyan raho: agar woh sirf meter ke peeche bhaage, toh shayad ek silly trick dhundh le jo meter ko bahut pasand ho lekin tum actually hate karo — isliye hum usse ek chhoti leash par rakhte hain sensible rehne ke liye.

Connections

  • RLHF — reward modeling iska middle stage hai (SFT → RM → PPO).
  • Bradley-Terry model — pairwise loss ki statistical foundation.
  • Logistic regression — score difference par same sigmoid-NLL structure.
  • PPO ko reward signal ke roop mein consume karta hai.
  • KL divergence — over-optimization rokne wali leash.
  • Goodhart's law — kyun proxy reward pressure mein degrade hota hai.
  • DPO Direct Preference Optimization — explicit RM skip karta hai same BT likelihood use karke.
  • Prompting aur RAG — reward train kiye bina behavior steer karne ke alternative tarike.

Concept Map

motivates

trains

cheap to collect

form

gives

scores yw and yl

negative log-likelihood

only difference matters

scalar score used in

gradient sigmoid of neg delta

Human preferences no formula

Reward model r-theta

Preference dataset x yw yl

Pairwise comparisons A beats B

Bradley-Terry model

P equals sigmoid of score diff

Pairwise RM loss

Push rw above rl

RLHF policy optimization

Gradient check