nutrition-label-llm
Un model de limbaj antrenat suplimentar care transformă textul OCR dezordonat de pe etichetele alimentare din UE în JSON curat cu valori nutriționale.
- Tip:
- Proiect personal
- Stare:
- Terminat
- An:
- 2026
Loc rezervat pentru imagine
Problema
OCR-ul pe etichete alimentare produce text zgomotos în multe limbi din UE, cu unități, separatori zecimali și denumiri sodiu sau sare diferite. Am construit modelul pentru funcția de scanare a etichetelor din versiunea 1 a Macro Micro.
Ce am construit
- Un set sintetic de antrenare de 50.000 de exemple, împărțit în 40.000 pentru antrenare, 5.000 pentru validare și 5.000 pentru testare.
- Scripturi de antrenare care ajustează Qwen2.5-14B-Instruct cu QLoRA pe 4 biți, pe un singur RTX 5090.
- Tratarea zecimalelor cu virgulă și cu punct, a variațiilor de unități, a diferențelor de limbă de pe etichete și a conversiei sodiu în sare (sodiu = sare / 2,5).
Tehnologii
- Python
- Qwen2.5-14B-Instruct
- Unsloth
- TRL
- QLoRA
Rezultate
- Un set de date sintetic de 50.000 de exemple și un proces complet de antrenare, publicate în depozit.