Improve model card for Variational Reasoning for Language Models
Browse filesThis PR significantly enhances the model card by:
* Adding `pipeline_tag: text-generation` to ensure discoverability for language models.
* Adding `license: apache-2.0`, inferred from the underlying frameworks (LLaMA-Factory and SkyThought) used in the repository.
* Including relevant `tags`: `llm`, `qwen`, and `reasoning` for better categorization and search.
* Updating the model card title to "Model Card for Variational Reasoning for Language Models".
* Populating the "Model Details" and "Model Sources" sections with comprehensive information from the paper and GitHub repository, including:
* A summary of the model based on the paper's abstract.
* Authors as developers.
* Model type, language, and finetuning details.
* Direct links to the paper: [Variational Reasoning for Language Models](https://huggingface.co/papers/2509.22637).
* Direct links to the GitHub repository: https://github.com/sail-sg/variational-reasoning for detailed usage, training, and evaluation instructions.
* Adding the BibTeX citation provided in the GitHub README.
Please review and merge these improvements to provide a more informative and discoverable model card.
|
@@ -1,199 +1,169 @@
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
-
tags:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
---
|
| 5 |
|
| 6 |
-
# Model Card for
|
| 7 |
-
|
| 8 |
-
<!-- Provide a quick summary of what the model is/does. -->
|
| 9 |
-
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
## Model Details
|
| 13 |
|
| 14 |
### Model Description
|
| 15 |
|
| 16 |
-
|
| 17 |
|
| 18 |
-
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
|
| 19 |
|
| 20 |
-
- **Developed by:**
|
| 21 |
-
- **
|
| 22 |
-
- **
|
| 23 |
-
- **
|
| 24 |
-
- **
|
| 25 |
-
- **License:** [More Information Needed]
|
| 26 |
-
- **Finetuned from model [optional]:** [More Information Needed]
|
| 27 |
|
| 28 |
-
### Model Sources
|
| 29 |
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
- **
|
| 33 |
-
- **Paper [optional]:** [More Information Needed]
|
| 34 |
-
- **Demo [optional]:** [More Information Needed]
|
| 35 |
|
| 36 |
## Uses
|
| 37 |
|
| 38 |
-
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
| 39 |
-
|
| 40 |
### Direct Use
|
| 41 |
|
| 42 |
-
|
| 43 |
-
|
| 44 |
-
[More Information Needed]
|
| 45 |
|
| 46 |
### Downstream Use [optional]
|
| 47 |
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
[More Information Needed]
|
| 51 |
|
| 52 |
### Out-of-Scope Use
|
| 53 |
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
[More Information Needed]
|
| 57 |
|
| 58 |
## Bias, Risks, and Limitations
|
| 59 |
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
[More Information Needed]
|
| 63 |
|
| 64 |
### Recommendations
|
| 65 |
|
| 66 |
-
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
| 67 |
-
|
| 68 |
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
| 69 |
|
| 70 |
## How to Get Started with the Model
|
| 71 |
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
[More Information Needed]
|
| 75 |
|
| 76 |
## Training Details
|
| 77 |
|
| 78 |
### Training Data
|
| 79 |
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
[More Information Needed]
|
| 83 |
|
| 84 |
### Training Procedure
|
| 85 |
|
| 86 |
-
|
| 87 |
|
| 88 |
#### Preprocessing [optional]
|
| 89 |
|
| 90 |
-
|
| 91 |
-
|
| 92 |
|
| 93 |
#### Training Hyperparameters
|
| 94 |
|
| 95 |
-
|
| 96 |
|
| 97 |
-
|
| 98 |
|
| 99 |
-
|
| 100 |
|
| 101 |
-
|
| 102 |
|
| 103 |
## Evaluation
|
| 104 |
|
| 105 |
-
<!-- This section describes the evaluation protocols and provides the results. -->
|
| 106 |
-
|
| 107 |
### Testing Data, Factors & Metrics
|
| 108 |
|
| 109 |
-
|
| 110 |
|
| 111 |
-
|
| 112 |
|
| 113 |
-
|
| 114 |
|
| 115 |
#### Factors
|
| 116 |
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
[More Information Needed]
|
| 120 |
|
| 121 |
#### Metrics
|
| 122 |
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
[More Information Needed]
|
| 126 |
|
| 127 |
### Results
|
| 128 |
|
| 129 |
-
|
| 130 |
|
| 131 |
#### Summary
|
| 132 |
|
| 133 |
-
|
| 134 |
|
| 135 |
## Model Examination [optional]
|
| 136 |
|
| 137 |
-
|
| 138 |
-
|
| 139 |
-
[More Information Needed]
|
| 140 |
|
| 141 |
## Environmental Impact
|
| 142 |
|
| 143 |
-
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
| 144 |
-
|
| 145 |
Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
|
| 146 |
|
| 147 |
-
- **Hardware Type:**
|
| 148 |
-
- **Hours used:**
|
| 149 |
-
- **Cloud Provider:**
|
| 150 |
-
- **Compute Region:**
|
| 151 |
-
- **Carbon Emitted:**
|
| 152 |
|
| 153 |
## Technical Specifications [optional]
|
| 154 |
|
| 155 |
### Model Architecture and Objective
|
| 156 |
|
| 157 |
-
|
| 158 |
|
| 159 |
### Compute Infrastructure
|
| 160 |
|
| 161 |
-
|
| 162 |
|
| 163 |
#### Hardware
|
| 164 |
|
| 165 |
-
|
| 166 |
|
| 167 |
#### Software
|
| 168 |
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
## Citation [optional]
|
| 172 |
-
|
| 173 |
-
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
| 174 |
|
| 175 |
-
|
| 176 |
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 182 |
|
| 183 |
## Glossary [optional]
|
| 184 |
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
[More Information Needed]
|
| 188 |
|
| 189 |
## More Information [optional]
|
| 190 |
|
| 191 |
-
|
| 192 |
|
| 193 |
## Model Card Authors [optional]
|
| 194 |
|
| 195 |
-
|
| 196 |
|
| 197 |
## Model Card Contact
|
| 198 |
|
| 199 |
-
[
|
|
|
|
| 1 |
---
|
| 2 |
library_name: transformers
|
| 3 |
+
tags:
|
| 4 |
+
- llm
|
| 5 |
+
- qwen
|
| 6 |
+
- reasoning
|
| 7 |
+
pipeline_tag: text-generation
|
| 8 |
+
license: apache-2.0
|
| 9 |
---
|
| 10 |
|
| 11 |
+
# Model Card for Variational Reasoning for Language Models
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
+
This model is associated with the paper [Variational Reasoning for Language Models](https://huggingface.co/papers/2509.22637), which introduces a framework for improving the reasoning ability of language models.
|
| 14 |
|
| 15 |
## Model Details
|
| 16 |
|
| 17 |
### Model Description
|
| 18 |
|
| 19 |
+
This repository contains models developed within a variational reasoning framework for language models that treats thinking traces as latent variables and optimizes them through variational inference. The framework extends the evidence lower bound (ELBO) to a multi-trace objective and proposes a forward-KL formulation for stable training of the variational posterior. It unifies variational inference with RL-style methods, yielding stable objectives for enhancing the reasoning capabilities of language models, as empirically validated on the Qwen 2.5 and Qwen 3 model families across various reasoning tasks.
|
| 20 |
|
| 21 |
+
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated and subsequently improved.
|
| 22 |
|
| 23 |
+
- **Developed by:** Xiangxin Zhou, Zichen Liu, Haonan Wang, Chao Du, Min Lin, Chongxuan Li, Liang Wang, Tianyu Pang
|
| 24 |
+
- **Model type:** Qwen3-based Language Model
|
| 25 |
+
- **Language(s) (NLP):** English
|
| 26 |
+
- **License:** Apache 2.0
|
| 27 |
+
- **Finetuned from model:** Qwen 2.5 and Qwen 3 model families
|
|
|
|
|
|
|
| 28 |
|
| 29 |
+
### Model Sources
|
| 30 |
|
| 31 |
+
- **Repository:** [https://github.com/sail-sg/variational-reasoning](https://github.com/sail-sg/variational-reasoning)
|
| 32 |
+
- **Paper:** [https://huggingface.co/papers/2509.22637](https://huggingface.co/papers/2509.22637)
|
| 33 |
+
- **Demo [optional]:** More Information Needed
|
|
|
|
|
|
|
| 34 |
|
| 35 |
## Uses
|
| 36 |
|
|
|
|
|
|
|
| 37 |
### Direct Use
|
| 38 |
|
| 39 |
+
These models are intended for text generation and advanced reasoning tasks, building upon their Qwen-based backbones. They can be integrated into applications requiring improved logical inference and problem-solving capabilities from language models.
|
|
|
|
|
|
|
| 40 |
|
| 41 |
### Downstream Use [optional]
|
| 42 |
|
| 43 |
+
More Information Needed
|
|
|
|
|
|
|
| 44 |
|
| 45 |
### Out-of-Scope Use
|
| 46 |
|
| 47 |
+
The models are not intended for generating harmful content, promoting misinformation, or engaging in any activities that violate ethical AI principles. They are specifically designed for reasoning tasks and may not perform optimally for tasks outside this scope without further fine-tuning.
|
|
|
|
|
|
|
| 48 |
|
| 49 |
## Bias, Risks, and Limitations
|
| 50 |
|
| 51 |
+
Please refer to the original Qwen backbone model cards for inherent biases and limitations. As these models are fine-tuned versions, they may inherit and potentially amplify certain biases present in their base models and training data. Users should exercise caution and conduct their own evaluations for specific applications.
|
|
|
|
|
|
|
| 52 |
|
| 53 |
### Recommendations
|
| 54 |
|
|
|
|
|
|
|
| 55 |
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
| 56 |
|
| 57 |
## How to Get Started with the Model
|
| 58 |
|
| 59 |
+
For detailed instructions on how to use, train, and evaluate these models within the Variational Reasoning framework, please refer to the official GitHub repository: [https://github.com/sail-sg/variational-reasoning](https://github.com/sail-sg/variational-reasoning). The repository provides comprehensive pipelines for training and evaluation.
|
|
|
|
|
|
|
| 60 |
|
| 61 |
## Training Details
|
| 62 |
|
| 63 |
### Training Data
|
| 64 |
|
| 65 |
+
For detailed information regarding the training data, please refer to the [training section](https://github.com/sail-sg/variational-reasoning#training) of the official GitHub repository.
|
|
|
|
|
|
|
| 66 |
|
| 67 |
### Training Procedure
|
| 68 |
|
| 69 |
+
This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure.
|
| 70 |
|
| 71 |
#### Preprocessing [optional]
|
| 72 |
|
| 73 |
+
More Information Needed
|
|
|
|
| 74 |
|
| 75 |
#### Training Hyperparameters
|
| 76 |
|
| 77 |
+
For detailed information regarding training hyperparameters and procedure, please refer to the [training section](https://github.com/sail-sg/variational-reasoning#training) of the official GitHub repository.
|
| 78 |
|
| 79 |
+
- **Training regime:** More Information Needed <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
| 80 |
|
| 81 |
+
#### Speeds, Sizes, Times [optional]
|
| 82 |
|
| 83 |
+
More Information Needed
|
| 84 |
|
| 85 |
## Evaluation
|
| 86 |
|
|
|
|
|
|
|
| 87 |
### Testing Data, Factors & Metrics
|
| 88 |
|
| 89 |
+
For detailed evaluation protocols and results, please refer to the [evaluation section](https://github.com/sail-sg/variational-reasoning#evaluation) of the official GitHub repository.
|
| 90 |
|
| 91 |
+
#### Testing Data
|
| 92 |
|
| 93 |
+
More Information Needed
|
| 94 |
|
| 95 |
#### Factors
|
| 96 |
|
| 97 |
+
More Information Needed
|
|
|
|
|
|
|
| 98 |
|
| 99 |
#### Metrics
|
| 100 |
|
| 101 |
+
More Information Needed
|
|
|
|
|
|
|
| 102 |
|
| 103 |
### Results
|
| 104 |
|
| 105 |
+
More Information Needed
|
| 106 |
|
| 107 |
#### Summary
|
| 108 |
|
| 109 |
+
More Information Needed
|
| 110 |
|
| 111 |
## Model Examination [optional]
|
| 112 |
|
| 113 |
+
More Information Needed
|
|
|
|
|
|
|
| 114 |
|
| 115 |
## Environmental Impact
|
| 116 |
|
|
|
|
|
|
|
| 117 |
Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
|
| 118 |
|
| 119 |
+
- **Hardware Type:** More Information Needed
|
| 120 |
+
- **Hours used:** More Information Needed
|
| 121 |
+
- **Cloud Provider:** More Information Needed
|
| 122 |
+
- **Compute Region:** More Information Needed
|
| 123 |
+
- **Carbon Emitted:** More Information Needed
|
| 124 |
|
| 125 |
## Technical Specifications [optional]
|
| 126 |
|
| 127 |
### Model Architecture and Objective
|
| 128 |
|
| 129 |
+
More Information Needed
|
| 130 |
|
| 131 |
### Compute Infrastructure
|
| 132 |
|
| 133 |
+
More Information Needed
|
| 134 |
|
| 135 |
#### Hardware
|
| 136 |
|
| 137 |
+
More Information Needed
|
| 138 |
|
| 139 |
#### Software
|
| 140 |
|
| 141 |
+
More Information Needed
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
|
| 143 |
+
## Citation
|
| 144 |
|
| 145 |
+
If you find this code useful, please consider citing our paper:
|
| 146 |
+
```bib
|
| 147 |
+
@article{zhou2025variationalreasoninglanguagemodels,
|
| 148 |
+
title={Variational Reasoning for Language Models},
|
| 149 |
+
author={Xiangxin Zhou and Zichen Liu and Haonan Wang and Chao Du and Min Lin and Chongxuan Li and Liang Wang and Tianyu Pang},
|
| 150 |
+
journal={arXiv preprint arXiv:2509.22637},
|
| 151 |
+
year={2025}
|
| 152 |
+
}
|
| 153 |
+
```
|
| 154 |
|
| 155 |
## Glossary [optional]
|
| 156 |
|
| 157 |
+
More Information Needed
|
|
|
|
|
|
|
| 158 |
|
| 159 |
## More Information [optional]
|
| 160 |
|
| 161 |
+
More Information Needed
|
| 162 |
|
| 163 |
## Model Card Authors [optional]
|
| 164 |
|
| 165 |
+
More Information Needed
|
| 166 |
|
| 167 |
## Model Card Contact
|
| 168 |
|
| 169 |
+
For any questions or concerns regarding this model, please refer to the contact information provided in the [GitHub repository](https://github.com/sail-sg/variational-reasoning).
|