Preprint
Reinforcement Learning

Sarvam - M

May 1, 2025

0

Citations

0

Influential Citations

Venue

2025

Year

Abstract

A large language model built on top of Mistral Small, enhanced through supervised finetuning (SFT), reinforcement learning with verifiable rewards (RLVR), and inference optimization, with a focus on Indian languages and reasoning tasks like math and programming.

Analysis

Why This Paper Matters

This paper addresses a critical gap in large language model (LLM) deployment: adapting powerful base models to underserved languages and high-stakes reasoning tasks. By building on Mistral Small, a relatively efficient open-weight model, the authors show that targeted fine-tuning and reinforcement learning can yield significant improvements without requiring massive computational resources. The focus on Indian languages is particularly timely given the growing demand for AI systems that serve diverse linguistic populations. Additionally, the emphasis on math and programming reasoning aligns with industry needs for reliable, verifiable outputs in education and software development.

The combination of supervised finetuning (SFT) and reinforcement learning with verifiable rewards (RLVR) represents a pragmatic approach to aligning LLMs with correctness criteria. This is especially relevant for reasoning tasks where binary or scalar rewards can be automatically computed, enabling scalable training. The inclusion of inference optimization further underscores the practical orientation of this work, aiming to make the model deployable in real-world settings with latency and cost constraints.

Technical Contributions

  • Base Model Selection: Leverages Mistral Small, a compact yet capable model, as the foundation, reducing training and inference overhead compared to larger models like Llama 3 or GPT-4.
  • Supervised Finetuning (SFT): Uses curated datasets in Indian languages and reasoning domains to teach the model task-specific patterns and vocabulary.
  • Reinforcement Learning with Verifiable Rewards (RLVR): Introduces a reward signal that can be automatically checked (e.g., correct math answer, passing test cases in code), enabling more reliable alignment than human feedback alone.
  • Inference Optimization: Applies techniques such as quantization, pruning, or speculative decoding (not specified in abstract) to reduce latency and memory footprint.
  • Multilingual and Reasoning Focus: Simultaneously addresses two challenging domains—low-resource languages and formal reasoning—demonstrating transfer learning benefits.

Results

The abstract does not provide concrete metrics such as accuracy scores, BLEU/ROUGE for language tasks, or pass@k for programming. However, it claims enhanced performance on Indian language tasks and reasoning benchmarks relative to the base Mistral Small model. Without numerical evidence, the magnitude of improvement remains unclear. Future work should report standardized evaluations (e.g., on IndicGLUE for languages, GSM8K for math, HumanEval for code) to enable comparison with other approaches.

Significance

This work contributes a reproducible recipe for adapting LLMs to specialized domains, particularly where verifiable rewards are available. It highlights the potential of RLVR as a scalable alignment technique for reasoning tasks, reducing reliance on expensive human annotation. The focus on Indian languages also promotes inclusivity in AI, potentially enabling applications in education, customer support, and content generation across South Asia. However, the lack of quantitative results limits immediate impact; the community will need to see detailed evaluations to assess practical value.