अनुक्रम से अनुक्रम मॉडल
Type: Build
Languages: Python
Prerequisites: Phase 5 · 08 (CNNs + RNNs for Text), Phase 3 · 11 (PyTorch Intro)
Time: ~75 minutes
समस्या
वर्गीकरण एक एकल लेबल पर एक चर लंबाई अनुक्रम का नक्शा बनाता है। अनुवाद एक चर लंबाई अनुक्रम को एक अन्य चर लंबाई अनुक्रम पर नक्शा बनाता है। इनपुट और आउटपुट अलग-अलग शब्दावली में रहते हैं, संभवतः अलग-अलग भाषाओं में, लंबाई समानता की कोई गारंटी नहीं है।
सीक्वॉक्स आर्किटेक्चर (Sutskever, Vinyals, Le, 2014) ने इसे एक जानबूझकर सरल नुस्खा के साथ क्रैक किया। दो आरएनएन। एक स्रोत वाक्य पढ़ता है और एक निश्चित आकार का संदर्भ वेक्टर उत्पन्न करता है। दूसरा उस वेक्टर को पढ़ता है और लक्ष्य वाक्य टोकन को टोकन द्वारा उत्पन्न करता है। वही कोड जिसे आपने पाठ 08 के लिए लिखा था, अलग तरह से चिपकाया गया।
यह दो कारणों से अध्ययन करने योग्य है। पहला, संदर्भ-वेक्टर बोतल की गर्दन एनएलपी में सबसे शैक्षणिक रूप से उपयोगी विफलता है। यह ध्यान और ट्रांसफार्मर में अच्छे सभी चीजों को प्रेरित करता है। दूसरा, प्रशिक्षण नुस्खा (शिक्षक मजबूर करना, अनुसूचित नमूने निकालना, निष्कर्ष पर बीम खोज) अभी भी एलएलएम सहित हर आधुनिक पीढ़ी प्रणाली पर लागू होता है।
अवधारणा
Encoder.एक आरएनएन जो स्रोत वाक्य पढ़ता है. इसकी अंतिम छिपी हुई स्थिति हैcontext vector पूरे इनपुट का एक निश्चित आकार का सारांश। स्रोत के अलावा कुछ भी नहीं खोना, माना जाता है।
Decoder.एक और आरएनएन संदर्भ वेक्टर से शुरू होता है। प्रत्येक चरण में यह पहले उत्पन्न टोकन को इनपुट के रूप में लेता है और लक्ष्य शब्दावली पर एक वितरण का उत्पादन करता है। अगले टोकन को चुनने के लिए नमूना या argmax। इसे वापस फ़ीड करें। एक तक दोहराएं।<EOS>टोकन उत्पन्न किया जाता है या अधिकतम लंबाई मारा जाता है।
Training:प्रत्येक डिकोडर चरण में क्रॉस-एंट्रोपी हानि, क्रम में योग। दोनों नेटवर्क के माध्यम से समय के माध्यम से मानक बैकपॉड।
Teacher forcing.प्रशिक्षण के दौरान, डेकोडर का इनपुट चरण में होता है tस्थिति पर ground-truth टोकन है t-1, डिकोडर की अपनी पिछली भविष्यवाणी नहीं है। यह प्रशिक्षण को स्थिर करता है; इसके बिना, प्रारंभिक त्रुटियां कैस्केड होती हैं और मॉडल कभी नहीं सीखता है। निष्कर्ष पर, आपको मॉडल की अपनी भविष्यवाणी का उपयोग करना होगा, इसलिए हमेशा एक ट्रेन / इन्फेरेंस वितरण अंतर होता है। उस अंतर को कहा जाता है exposure bias. .
The bottleneck.कोडर को स्रोत के बारे में जो कुछ भी सीखा है उसे उस संदर्भ वेक्टर में दबाया जाना चाहिए। लंबे वाक्य विवरण खो देते हैं। दुर्लभ शब्द धुंधला हो जाते हैं। पुनर्गठन (चैट नोअर बनाम ब्लैक कैट) को याद रखना चाहिए, गणना नहीं।
ध्यान (पाठ 10) यह ठीक करता है कि डिकोडर को प्रत्येक को देखने दे एन्कोडर छिपे हुए राज्य, न केवल अंतिम। यह पूरी पिच है।
इसे बनाओ
चरण 1: एक एन्कोडर
pythonimport torch
import torch.nn as nn
class Encoder(nn.Module):
def __init__(self, src_vocab_size, embed_dim, hidden_dim):
super().__init__()
self.embed = nn.Embedding(src_vocab_size, embed_dim, padding_idx=0)
self.gru = nn.GRU(embed_dim, hidden_dim, batch_first=True)
def forward(self, src):
e = self.embed(src)
outputs, hidden = self.gru(e)
return outputs, hiddenoutputsआकार है[batch, seq_len, hidden_dim] प्रत्येक इनपुट स्थिति के लिए एक छिपा हुआ राज्य। hiddenआकार है[1, batch, hidden_dim] अंतिम चरण। पाठ 08 में कहा गया था "वर्गीकरण के लिए आउटपुट परpool।" यहाँ हम अंतिम छिपे हुए राज्य को संदर्भ वेक्टर के रूप में रखते हैं, और प्रति चरण आउटपुट को अनदेखा करते हैं।
चरण 2: एक डिकोडर
pythonclass Decoder(nn.Module):
def __init__(self, tgt_vocab_size, embed_dim, hidden_dim):
super().__init__()
self.embed = nn.Embedding(tgt_vocab_size, embed_dim, padding_idx=0)
self.gru = nn.GRU(embed_dim, hidden_dim, batch_first=True)
self.fc = nn.Linear(hidden_dim, tgt_vocab_size)
def forward(self, token, hidden):
e = self.embed(token)
out, hidden = self.gru(e, hidden)
logits = self.fc(out)
return logits, hiddenइनपुटः एकल टोकन का एक बैच और वर्तमान छिपी हुई स्थिति। आउटपुटः अगले टोकन और अद्यतन छिपी हुई स्थिति के लिए शब्दावली लॉगिंग।
चरण 3: शिक्षक द्वारा मजबूर किए जाने वाले प्रशिक्षण लूप
pythondef train_batch(encoder, decoder, src, tgt, bos_id, optimizer, teacher_forcing_ratio=0.9):
optimizer.zero_grad()
_, hidden = encoder(src)
batch_size, tgt_len = tgt.shape
input_token = torch.full((batch_size, 1), bos_id, dtype=torch.long)
loss = 0.0
loss_fn = nn.CrossEntropyLoss(ignore_index=0)
for t in range(tgt_len):
logits, hidden = decoder(input_token, hidden)
step_loss = loss_fn(logits.squeeze(1), tgt[:, t])
loss += step_loss
use_teacher = torch.rand(1).item() < teacher_forcing_ratio
if use_teacher:
input_token = tgt[:, t].unsqueeze(1)
else:
input_token = logits.argmax(dim=-1)
loss.backward()
optimizer.step()
return loss.item() / tgt_lenदो बटन नाम देने लायक हैं।ignore_index=0पैडिंग टोकन पर नुकसान छोड़ देता है। teacher_forcing_ratioप्रत्येक चरण में मॉडल की भविष्यवाणी के खिलाफ वास्तविक टोकन का उपयोग करने की संभावना है। एक्सपोज़र-bias अंतर को बंद करने के लिए 1.0 (पूर्ण शिक्षक मजबूर) से शुरू करें और प्रशिक्षण के माध्यम से ~0.5 तक नीचे बढ़ें।
चरण 4: निष्कर्ष लूप (लाभकारी)
python@torch.no_grad()
def greedy_decode(encoder, decoder, src, bos_id, eos_id, max_len=50):
_, hidden = encoder(src)
batch_size = src.shape[0]
input_token = torch.full((batch_size, 1), bos_id, dtype=torch.long)
output_ids = []
for _ in range(max_len):
logits, hidden = decoder(input_token, hidden)
next_token = logits.argmax(dim=-1)
output_ids.append(next_token)
input_token = next_token
if (next_token == eos_id).all():
break
return torch.cat(output_ids, dim=1)लालची डिकोडिंग हर कदम पर सबसे अधिक संभावना वाले टोकन को चुनती है। यह भटक सकता हैः एक बार जब आप एक टोकन के लिए प्रतिबद्ध होते हैं, तो आप इसे अनलॉक नहीं कर सकते। Beam searchशीर्ष को बनाए रखता है-kआंशिक अनुक्रमों को जीवित और सबसे अधिक स्कोरिंग पूर्ण एक को चुनता है अंत में. बीम चौड़ाई 3-5 मानक है।
चरण 5: बोतल की गर्दन, दिखाया गया
मॉडल को खिलौना कॉपी करने के लिए प्रशिक्षित करेंः स्रोत [a, b, c, d, e], लक्ष्य [a, b, c, d, e]अनुक्रम की लंबाई बढ़ाएँ। सटीकता का निरीक्षण करें।
seq_len=5 copy accuracy: 98%
seq_len=10 copy accuracy: 91%
seq_len=20 copy accuracy: 62%
seq_len=40 copy accuracy: 23%एक ही GRU छिपी हुई स्थिति 40 टोकन इनपुट को याद नहीं रख सकती है। जानकारी को प्रत्येक एन्कोडर चरण में वहां रखा जाता है, लेकिन डिकोडर केवल अंतिम स्थिति देखता है। ध्यान इसे सीधे ठीक करता है।
इसका प्रयोग करें
पायटॉर्च ने nn.Transformerऔर nn.LSTM-आधारित अनुक्रम टेम्पलेट्स.transformersपुस्तकालय जहाजों को पूर्ण एन्कोडर-डेकोडर मॉडल (BART, T5, mBART, NLLB) के साथ प्रशिक्षित किया गया है अरबों टोकन.
pythonfrom transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tok = AutoTokenizer.from_pretrained("facebook/bart-base")
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-base")
src = tok("Translate this to French: Hello, how are you?", return_tensors="pt")
out = model.generate(**src, max_new_tokens=50, num_beams=4)
print(tok.decode(out[0], skip_special_tokens=True))आधुनिक एन्कोडर-डेकोडरों ने ट्रांसफार्मर के लिए आरएनएन छोड़ दिया। उच्च-स्तरीय आकार (एन्कोडर, डेकोडर, जेनरेट-टोकन-टू-टोकन) 2014 के सीक्डब्ल्यूसी पेपर के समान है। प्रत्येक ब्लॉक के अंदर तंत्र अलग है।
RNN आधारित seq2seq के लिए अभी तक कब पहुंचना है
नए परियोजनाओं के लिए लगभग कभी नहीं।
- स्ट्रीमिंग अनुवाद जहां आप सीमित स्मृति के साथ एक समय में एक टोकन इनपुट का उपभोग करते हैं।
- डिवाइस पर पाठ उत्पन्न करना जहां ट्रांसफार्मर मेमोरी की लागत निषेधात्मक है।
- शैक्षणिक. एन्कोडर-डेकोडर की गड़बड़ी को समझना यह समझने का सबसे तेज़ तरीका है कि ट्रांसफार्मर क्यों जीते हैं।
जोखिम पूर्वाग्रह और इसके उन्मूलन
- Scheduled sampling.प्रशिक्षण के दौरान शिक्षक बल अनुपात के साथ-साथ, मॉडल अपनी गलतियों से उबरना सीखता है।
- Minimum risk training.वाक्य स्तर के ब्लू स्कोर पर प्रशिक्षित करें टोकन स्तर के क्रॉस-एंट्रोपी के बजाय. आप वास्तव में क्या चाहते हैं के करीब.
- Reinforcement learning fine-tuning.आधुनिक LLM RLHF में इस्तेमाल एक मीट्रिक के साथ अनुक्रम जनरेटर इनाम.
तीनों ही अभी भी ट्रांसफार्मर आधारित पीढ़ी पर लागू होते हैं।
इसे भेजें
outputs/prompt-seq2seq-design.md:
markdown---
name: seq2seq-design
description: Design a sequence-to-sequence pipeline for a given task.
phase: 5
lesson: 09
---
Given a task (translation, summarization, paraphrase, question rewrite), output:
1. Architecture. Pretrained transformer encoder-decoder (BART, T5, mBART, NLLB) is the default. RNN-based seq2seq only for specific constraints.
2. Starting checkpoint. Name it (`facebook/bart-base`, `google/flan-t5-base`, `facebook/nllb-200-distilled-600M`). Match the checkpoint to task and language coverage.
3. Decoding strategy. Greedy for deterministic output, beam search (width 4-5) for quality, sampling with temperature for diversity. One sentence justification.
4. One failure mode to verify before shipping. Exposure bias manifests as generation drift on longer outputs; sample 20 outputs at the 90th-percentile length and eyeball.
Refuse to recommend training a seq2seq from scratch for under a million parallel examples. Flag any pipeline that uses greedy decoding for user-facing content as fragile (greedy repeats and loops).व्यायाम
- Easy.खेलौना कॉपी कार्य को लागू करें। इनपुट-आउटपुट जोड़े पर एक GRU अनुक्रम को प्रशिक्षित करें जहां लक्ष्य स्रोत के बराबर है। लंबाई 5, 10, 20 पर सटीकता मापें। बोतल गला को पुनः उत्पन्न करें।
- Medium.बीम सर्च को बीम चौड़ाई के साथ डिकोडिंग जोड़ें 3. लालच के खिलाफ एक छोटे समानांतर कॉर्पस पर ब्लू मापें। दस्तावेज़ जहां बीम सर्च जीतता है (आमतौर पर अंतिम टोकन) और जहां यह कोई फर्क नहीं पड़ता है।
- Hard.ठीक-ठीक
facebook/bart-base10k-pair पैराफ्रेस डेटासेट पर। ठीक-ट्यून किए गए मॉडल के बीम-4 आउटपुट की तुलना मूल मॉडल के लिए रखे गए इनपुट पर करें। BLEU रिपोर्ट करें और 10 गुणात्मक उदाहरण चुनें।
प्रमुख शर्तें
| Term | What people say | What it actually means |
|---|---|---|
| Encoder | Input RNN | Reads source. Produces per-step hidden states and a final context vector. |
| Decoder | Output RNN | Initialized from context vector. Generates target tokens one at a time. |
| Context vector | The summary | Final encoder hidden state. Fixed size. The bottleneck attention solves. |
| Teacher forcing | Use true tokens | Feed the ground-truth previous token at training time. Stabilizes learning. |
| Exposure bias | Train/test gap | Model trained on true tokens never practiced recovering from its own mistakes. |
| Beam search | Better decoding | Keep top-k partial sequences alive at each step instead of committing greedily. |
आगे पढ़ना
- Sutskever, Vinyals, Le (2014). Sequence to Sequence Learning with Neural Networks मूल अनुक्रमिक कागज। चार पृष्ठ।
- Cho et al. (2014). Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation जीआरयू और एन्कोडर-डेकोडर फ्रेमिंग की शुरुआत की गई।
- Bahdanau, Cho, Bengio (2014). Neural Machine Translation by Jointly Learning to Align and Translate ध्यान पत्र. इस पाठ के तुरंत बाद पढ़ें।
- PyTorch NLP from Scratch tutorial निर्माण योग्य अनुक्रम + ध्यान कोड।
This free lesson is part of the AI Engineering from Scratch curriculum. Read the full explanation, run the lesson code, and verify the result in the interactive reader or from the repository source.
Browse the complete course catalog or open this lesson on GitHub.