Update README.md
Browse files
README.md
CHANGED
|
@@ -64,15 +64,36 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 64 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 65 |
outputs = model.generate(**inputs, max_new_tokens=1200)
|
| 66 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
|
|
|
| 67 |
|
| 68 |
|
|
|
|
| 69 |
|
| 70 |
-
|
| 71 |
-
### If you use this model or method in your research, please cite our paper:
|
| 72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
@article{thaker2025knowledge,
|
| 74 |
title={Knowledge Distillation with Structured Chain-of-Thought for Text-to-SQL},
|
| 75 |
author={Thaker, Khushboo and Bresler, Yony},
|
| 76 |
journal={arXiv preprint arXiv:2512.17053},
|
| 77 |
year={2025}
|
| 78 |
-
}
|
|
|
|
|
|
| 64 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 65 |
outputs = model.generate(**inputs, max_new_tokens=1200)
|
| 66 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
| 67 |
+
```
|
| 68 |
|
| 69 |
|
| 70 |
+
## Intended Use
|
| 71 |
|
| 72 |
+
Struct-SQL-4B is intended for **research and academic use** in tasks involving **Text-to-SQL generation** and **semantic parsing over relational databases**. The model is particularly suited for studying:
|
|
|
|
| 73 |
|
| 74 |
+
- Knowledge distillation techniques that leverage **structured intermediate representations**
|
| 75 |
+
- Explicit **query planning** as an alternative to unstructured chain-of-thought reasoning
|
| 76 |
+
- Error reduction in SQL generation, including syntactic validity and schema grounding
|
| 77 |
+
- Compact language models for complex reasoning under limited parameter budgets
|
| 78 |
+
|
| 79 |
+
The model is not optimized for direct deployment in production database systems without additional validation and safety constraints.
|
| 80 |
+
|
| 81 |
+
---
|
| 82 |
+
|
| 83 |
+
## Limitations
|
| 84 |
+
|
| 85 |
+
- Evaluation is confined to the SQLite-based BIRD benchmark
|
| 86 |
+
- The model may generate logically plausible but incorrect SQL for highly complex multi-hop queries
|
| 87 |
+
|
| 88 |
+
---
|
| 89 |
+
|
| 90 |
+
## Citation
|
| 91 |
+
|
| 92 |
+
```bibtex
|
| 93 |
@article{thaker2025knowledge,
|
| 94 |
title={Knowledge Distillation with Structured Chain-of-Thought for Text-to-SQL},
|
| 95 |
author={Thaker, Khushboo and Bresler, Yony},
|
| 96 |
journal={arXiv preprint arXiv:2512.17053},
|
| 97 |
year={2025}
|
| 98 |
+
}
|
| 99 |
+
|