Update README.md

README.md

@@ -1,3 +1,117 @@
----
-license: mit
----
+---
+license: mit
+tags:
+- rebalance
+- steering-vector
+- reasoning
+- llm
+- iclr-2026
+language:
+- en
+---
+
+
+
+<h1 align="center">ReBalance Steering Vectors</h1>
+
+<p align="center">
+  Steering vectors for <strong>Efficient Reasoning with Balanced Thinking</strong> (ICLR 2026)
+</p>
+
+<p align="center">
+  <a href="https://huggingface.co/papers/2603.12372"><img src="https://img.shields.io/badge/Paper-Hugging_Face-b31b1b.svg" alt="Paper (Hugging Face)"></a>
+  <a href="https://openreview.net/forum?id=cJseWJJ5IM"><img src="https://img.shields.io/badge/Paper-Open_Review-8D1B12.svg" alt="Paper (Open Review)"></a>
+  <a href="https://github.com/yu-lin-li/ReBalance"><img src="https://img.shields.io/badge/Code-GitHub-black.svg" alt="Code (GitHub)"></a>
+  <a href="https://rebalance-ai.github.io"><img src="https://img.shields.io/badge/Project-Page-2EA44F.svg" alt="Project Page"></a>
+  <a href="https://github.com/yu-lin-li/ReBalance/blob/main/LICENSE"><img src="https://img.shields.io/badge/License-MIT-yellow.svg" alt="License: MIT"></a>
+</p>
+
+## Overview
+This repository provides **steering vectors only** for ReBalance.
+
+It does **not** include:
+- base model weights,
+- inference/training code,
+- datasets.
+
+Use this repository together with the official codebase:
+- Code: https://github.com/yu-lin-li/ReBalance
+- Inference script: `transformer_inference_steer_dp.py`
+
+## Available Steering Vectors
+| Base model | Vector path | Recommended `--steer_layer` |
+|---|---|---|
+| DeepSeek-R1-Distill-Qwen-1.5B | `vectors/DeepSeek-R1-Distill-Qwen-1.5B/steer_vector_layer19_conf_mixed.pt` | `19` |
+| DeepSeek-R1-Distill-Qwen-7B | `vectors/DeepSeek-R1-Distill-Qwen-7B/steer_vector_layer22_conf_mixed.pt` | `22` |
+| QwQ-32B | `vectors/QwQ-32B/steer_vector_layer58_conf_mixed.pt` | `58` |
+
+## Local Directory Layout (after download)
+```text
+ReBalance/
+├── transformer_inference_steer_dp.py
+└── vectors/
+    ├── DeepSeek-R1-Distill-Qwen-1.5B/
+    │   └── steer_vector_layer19_conf_mixed.pt
+    ├── DeepSeek-R1-Distill-Qwen-7B/
+    │   └── steer_vector_layer22_conf_mixed.pt
+    └── QwQ-32B/
+        └── steer_vector_layer58_conf_mixed.pt
+```
+
+## Download
+### Option 1: Clone the full model repository
+```bash
+git lfs install
+git clone https://huggingface.co/Yulin-Li/ReBalance
+```
+
+Then copy the `vectors/` folder into your local `ReBalance/` root directory.
+
+### Option 2: Download only the vectors with `huggingface_hub`
+```python
+from huggingface_hub import snapshot_download
+
+snapshot_download(
+    repo_id="Yulin-Li/ReBalance",
+    repo_type="model",
+    allow_patterns="vectors/*",
+    local_dir="."
+)
+```
+
+## Quick Usage with ReBalance
+```bash
+python transformer_inference_steer_dp.py \
+  --model_name_or_path 'DeepSeek-R1-Distill-Qwen-1.5B' \
+  --dataset_dir "./Data/" \
+  --output_path "./outputs" \
+  --dataset "Math_AIME2024" \
+  --max_generated_tokens 16000 \
+  --num_gpus 8 \
+  --steer_vector_path ./vectors/DeepSeek-R1-Distill-Qwen-1.5B/steer_vector_layer19_conf_mixed.pt \
+  --steer_layer 19 \
+  --steer_coef -1
+```
+
+## Paper, Code, and Project Links
+- Paper (Hugging Face): https://huggingface.co/papers/2603.12372
+- Paper (Open Review): https://openreview.net/forum?id=cJseWJJ5IM
+- Code: https://github.com/yu-lin-li/ReBalance
+- Project page: https://rebalance-ai.github.io
+
+## Intended Use
+- Research and reproducibility for ReBalance.
+- Experiments on reasoning efficiency and accuracy trade-offs.
+- Comparative studies on overthinking mitigation.
+
+## Citation
+If you find ReBalance useful in your research, please cite our paper:
+
+```bibtex
+@article{li2026efficient,
+  title={Efficient Reasoning with Balanced Thinking},
+  author={Li, Yulin and Tu, Tengyao and Ding, Li and Wang, Junjie and Zhen, Huiling and Chen, Yixin and Li, Yong and Tian, Zhuotao},
+  booktitle={Proceedings of the 14th International Conference on Learning Representations},
+  year={2026}
+}
+```