nielsr HF Staff commited on
Commit
1e85aab
·
verified ·
1 Parent(s): d4bb852

Improve model card for Variational Reasoning for Language Models

Browse files

This 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.

Files changed (1) hide show
  1. README.md +62 -92
README.md CHANGED
@@ -1,199 +1,169 @@
1
  ---
2
  library_name: transformers
3
- tags: []
 
 
 
 
 
4
  ---
5
 
6
- # Model Card for Model ID
7
-
8
- <!-- Provide a quick summary of what the model is/does. -->
9
-
10
 
 
11
 
12
  ## Model Details
13
 
14
  ### Model Description
15
 
16
- <!-- Provide a longer summary of what this model is. -->
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:** [More Information Needed]
21
- - **Funded by [optional]:** [More Information Needed]
22
- - **Shared by [optional]:** [More Information Needed]
23
- - **Model type:** [More Information Needed]
24
- - **Language(s) (NLP):** [More Information Needed]
25
- - **License:** [More Information Needed]
26
- - **Finetuned from model [optional]:** [More Information Needed]
27
 
28
- ### Model Sources [optional]
29
 
30
- <!-- Provide the basic links for the model. -->
31
-
32
- - **Repository:** [More Information Needed]
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
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
-
44
- [More Information Needed]
45
 
46
  ### Downstream Use [optional]
47
 
48
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
-
50
- [More Information Needed]
51
 
52
  ### Out-of-Scope Use
53
 
54
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
-
56
- [More Information Needed]
57
 
58
  ## Bias, Risks, and Limitations
59
 
60
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
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
- Use the code below to get started with the model.
73
-
74
- [More Information Needed]
75
 
76
  ## Training Details
77
 
78
  ### Training Data
79
 
80
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
-
82
- [More Information Needed]
83
 
84
  ### Training Procedure
85
 
86
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
 
88
  #### Preprocessing [optional]
89
 
90
- [More Information Needed]
91
-
92
 
93
  #### Training Hyperparameters
94
 
95
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
 
97
- #### Speeds, Sizes, Times [optional]
98
 
99
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
 
101
- [More Information Needed]
102
 
103
  ## Evaluation
104
 
105
- <!-- This section describes the evaluation protocols and provides the results. -->
106
-
107
  ### Testing Data, Factors & Metrics
108
 
109
- #### Testing Data
110
 
111
- <!-- This should link to a Dataset Card if possible. -->
112
 
113
- [More Information Needed]
114
 
115
  #### Factors
116
 
117
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
-
119
- [More Information Needed]
120
 
121
  #### Metrics
122
 
123
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
-
125
- [More Information Needed]
126
 
127
  ### Results
128
 
129
- [More Information Needed]
130
 
131
  #### Summary
132
 
133
-
134
 
135
  ## Model Examination [optional]
136
 
137
- <!-- Relevant interpretability work for the model goes here -->
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:** [More Information Needed]
148
- - **Hours used:** [More Information Needed]
149
- - **Cloud Provider:** [More Information Needed]
150
- - **Compute Region:** [More Information Needed]
151
- - **Carbon Emitted:** [More Information Needed]
152
 
153
  ## Technical Specifications [optional]
154
 
155
  ### Model Architecture and Objective
156
 
157
- [More Information Needed]
158
 
159
  ### Compute Infrastructure
160
 
161
- [More Information Needed]
162
 
163
  #### Hardware
164
 
165
- [More Information Needed]
166
 
167
  #### Software
168
 
169
- [More Information Needed]
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
- **BibTeX:**
176
 
177
- [More Information Needed]
178
-
179
- **APA:**
180
-
181
- [More Information Needed]
 
 
 
 
182
 
183
  ## Glossary [optional]
184
 
185
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
-
187
- [More Information Needed]
188
 
189
  ## More Information [optional]
190
 
191
- [More Information Needed]
192
 
193
  ## Model Card Authors [optional]
194
 
195
- [More Information Needed]
196
 
197
  ## Model Card Contact
198
 
199
- [More Information Needed]
 
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).