Learning from Generalization Patterns: An Evaluation-Driven Approach to Enhanced Data Augmentation for Fine-Tuning Small Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Song, Huan, Razdan, Deeksha, Qian, Yiyue, Chowdhury, Arijit Ghosh, Patwa, Parth, Chadha, Aman, Zhang, Shinan, Keshava, Sharlina, Marlowe, Hannah
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909860777426944
author Song, Huan
Razdan, Deeksha
Qian, Yiyue
Chowdhury, Arijit Ghosh
Patwa, Parth
Chadha, Aman
Zhang, Shinan
Keshava, Sharlina
Marlowe, Hannah
author_facet Song, Huan
Razdan, Deeksha
Qian, Yiyue
Chowdhury, Arijit Ghosh
Patwa, Parth
Chadha, Aman
Zhang, Shinan
Keshava, Sharlina
Marlowe, Hannah
contents Small Language Models (SLMs) offer compelling advantages in deployment cost and latency, but their accuracy often lags behind larger models, particularly for complex domain-specific tasks. While supervised fine-tuning can help bridge this performance gap, it requires substantial manual effort in data preparation and iterative optimization. We present PaDA-Agent (Pattern-guided Data Augmentation Agent), an evaluation-driven approach that streamlines the data augmentation process for SLMs through coordinated operations. Unlike state-of-the-art approaches that focus on model training errors only and generating error-correcting samples, PaDA-Agent discovers failure patterns from the validation data via evaluations and drafts targeted data augmentation strategies aiming to directly reduce the generalization gap. Our experimental results demonstrate significant improvements over state-of-the-art LLM-based data augmentation approaches for Llama 3.2 1B Instruct model fine-tuning.
format Preprint
id arxiv_https___arxiv_org_abs_2510_18143
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning from Generalization Patterns: An Evaluation-Driven Approach to Enhanced Data Augmentation for Fine-Tuning Small Language Models
Song, Huan
Razdan, Deeksha
Qian, Yiyue
Chowdhury, Arijit Ghosh
Patwa, Parth
Chadha, Aman
Zhang, Shinan
Keshava, Sharlina
Marlowe, Hannah
Artificial Intelligence
Machine Learning
Small Language Models (SLMs) offer compelling advantages in deployment cost and latency, but their accuracy often lags behind larger models, particularly for complex domain-specific tasks. While supervised fine-tuning can help bridge this performance gap, it requires substantial manual effort in data preparation and iterative optimization. We present PaDA-Agent (Pattern-guided Data Augmentation Agent), an evaluation-driven approach that streamlines the data augmentation process for SLMs through coordinated operations. Unlike state-of-the-art approaches that focus on model training errors only and generating error-correcting samples, PaDA-Agent discovers failure patterns from the validation data via evaluations and drafts targeted data augmentation strategies aiming to directly reduce the generalization gap. Our experimental results demonstrate significant improvements over state-of-the-art LLM-based data augmentation approaches for Llama 3.2 1B Instruct model fine-tuning.
title Learning from Generalization Patterns: An Evaluation-Driven Approach to Enhanced Data Augmentation for Fine-Tuning Small Language Models
topic Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2510.18143