136 lines
5.1 KiB
Markdown
136 lines
5.1 KiB
Markdown
---
|
|
library_name: transformers
|
|
license: apache-2.0
|
|
base_model: Qwen/Qwen3-0.6B
|
|
pipeline_tag: text-generation
|
|
language:
|
|
- bn
|
|
- en
|
|
tags:
|
|
- math
|
|
- bengali
|
|
- reasoning
|
|
- sft
|
|
datasets:
|
|
- dipta007/Ganit
|
|
---
|
|
|
|
# GanitLLM-0.6B_SFT
|
|
|
|
<p align="center">
|
|
<a href="https://arxiv.org/abs/2601.06767">
|
|
<img src="https://img.shields.io/badge/%F0%9F%94%A5_Accepted_at-ACL_2026_(Findings)_%F0%9F%94%A5-b12a00?style=for-the-badge&labelColor=ffb300" alt="Accepted at ACL 2026 (Findings)">
|
|
</a>
|
|
</p>
|
|
|
|
[](https://arxiv.org/abs/2601.06767)
|
|
[](https://arxiv.org/abs/2601.06767)
|
|
[](https://dipta007.github.io/GanitLLM/)
|
|
[](https://huggingface.co/datasets/dipta007/Ganit)
|
|
[](https://huggingface.co/collections/dipta007/ganitllm)
|
|
[](https://github.com/dipta007/GanitLLM)
|
|
|
|
## Highlights
|
|
|
|
**GanitLLM-0.6B_SFT** is our smallest Bengali mathematical reasoning model trained with Supervised Fine-Tuning on the GANIT dataset. Ideal for resource-constrained deployments. Key improvements over the base Qwen3-0.6B model:
|
|
|
|
- **+20.00 accuracy** on Bn-MGSM benchmark (8.40 → 28.40)
|
|
- **+39.20 accuracy** on Bn-MSVAMP benchmark (12.20 → 51.40)
|
|
- **88.60% Bengali reasoning** (vs 12.43% for base model)
|
|
- **79.2% fewer words** in generated solutions (1265 → 263 words)
|
|
|
|
> **Note**: This is the SFT-only checkpoint. For best results, use the RL-enhanced versions: [GanitLLM-0.6B_SFT_CGRPO](https://huggingface.co/dipta007/GanitLLM-0.6B_SFT_CGRPO) or [GanitLLM-0.6B_SFT_GRPO](https://huggingface.co/dipta007/GanitLLM-0.6B_SFT_GRPO).
|
|
|
|
## Model Overview
|
|
|
|
| Property | Value |
|
|
|----------|-------|
|
|
| **Model Type** | Causal Language Model |
|
|
| **Base Model** | Qwen/Qwen3-0.6B |
|
|
| **Parameters** | 0.6B |
|
|
| **Training** | Supervised Fine-Tuning |
|
|
| **Context Length** | 4,096 tokens |
|
|
| **Language** | Bengali, English |
|
|
|
|
## Training Details
|
|
|
|
This model was trained with a single-stage pipeline:
|
|
|
|
1. **Supervised Fine-Tuning (SFT)**: Trained on GANIT-SFT (~11k examples) to ground reasoning in Bengali
|
|
|
|
### Training Data
|
|
- **Dataset**: GANIT-SFT (11,023 examples)
|
|
- **Format**: Bengali math problems with chain-of-thought reasoning
|
|
- **Structure**: `<think>` tags for reasoning, `<answer>` tags for final answer
|
|
|
|
## Quickstart
|
|
|
|
```python
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
model_name = "dipta007/GanitLLM-0.6B_SFT"
|
|
|
|
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
model = AutoModelForCausalLM.from_pretrained(
|
|
model_name,
|
|
torch_dtype="auto",
|
|
device_map="auto"
|
|
)
|
|
|
|
problem = "একটি দোকানে ১২টি আপেল আছে। যদি ৫টি আপেল বিক্রি হয়, তাহলে কতটি আপেল বাকি থাকবে?"
|
|
|
|
prompt = f"""A conversation takes place between the user and the assistant. The user asks a question, and the assistant solves the problem. Please reason step by step in Bengali, and put your final answer in the <answer> </answer> tags.
|
|
|
|
Question: {problem}"""
|
|
|
|
messages = [{"role": "user", "content": prompt}]
|
|
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
|
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
|
|
|
generated_ids = model.generate(**model_inputs, max_new_tokens=2048, temperature=0.7)
|
|
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
|
response = tokenizer.decode(output_ids, skip_special_tokens=True)
|
|
print(response)
|
|
```
|
|
|
|
### Using vLLM
|
|
|
|
```bash
|
|
vllm serve dipta007/GanitLLM-0.6B_SFT --max-model-len 4096
|
|
```
|
|
|
|
## Performance
|
|
|
|
| Model | Bn-MGSM | Bn-MSVAMP | Avg. Words | Bengali % |
|
|
|-------|---------|-----------|------------|-----------|
|
|
| Qwen3-0.6B (base) | 8.40 | 12.20 | 1265 | 12.43% |
|
|
| **GanitLLM-0.6B_SFT** | **28.40** | **51.40** | **263** | **88.60%** |
|
|
|
|
## Related Models
|
|
|
|
| Model | Parameters | Training | Link |
|
|
|-------|------------|----------|------|
|
|
| GanitLLM-0.6B_SFT_CGRPO | 0.6B | SFT + CGRPO | [Link](https://huggingface.co/dipta007/GanitLLM-0.6B_SFT_CGRPO) |
|
|
| GanitLLM-0.6B_SFT_GRPO | 0.6B | SFT + GRPO | [Link](https://huggingface.co/dipta007/GanitLLM-0.6B_SFT_GRPO) |
|
|
| **GanitLLM-0.6B_SFT** | 0.6B | SFT | [Link](https://huggingface.co/dipta007/GanitLLM-0.6B_SFT) |
|
|
| GanitLLM-0.6B_CGRPO | 0.6B | CGRPO | [Link](https://huggingface.co/dipta007/GanitLLM-0.6B_CGRPO) |
|
|
|
|
## Citation
|
|
|
|
```bibtex
|
|
@inproceedings{dipta2026ganitllm,
|
|
title={GanitLLM: Difficulty-Aware Bengali Mathematical Reasoning through Curriculum-GRPO},
|
|
author={Shubhashis Roy Dipta and Khairul Mahbub and Nadia Najjar},
|
|
booktitle={Findings of the Association for Computational Linguistics: ACL 2026},
|
|
year={2026},
|
|
eprint={2601.06767},
|
|
archivePrefix={arXiv},
|
|
primaryClass={cs.CL},
|
|
url={https://arxiv.org/abs/2601.06767},
|
|
}
|
|
```
|
|
|
|
## License
|
|
|
|
This model is released under the Apache 2.0 License.
|