SStarrySSky commited on
Commit
ff73af5
·
verified ·
1 Parent(s): 09ef821

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +161 -3
README.md CHANGED
@@ -1,3 +1,161 @@
1
- ---
2
- license: gpl-3.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Askit-OLMo-32B-Spatial-Thinking-Preview
2
+
3
+ ## 🎯 Model Highlights
4
+
5
+ **Askit-OLMo-32B-Spatial-Thinking-Preview** is a fine-tuned version of OLMo 3.1 32B, specifically optimized for **spatial reasoning** and **physics simulation code generation**. This model excels at explicit 3D coordinate calculations and API-driven physics modeling.
6
+
7
+ ### 🧠 Core Innovation: Spatial Intuition + API Fusion
8
+
9
+ Unlike traditional code generation models, Askit enforces **explicit spatial thinking** in its reasoning chain:
10
+
11
+ - **Forced Spatial Cognition**: Every physics problem is decomposed into 3D coordinate calculations
12
+ - **API-Centric Reasoning**: Thinking chains explicitly translate physics → PhysicsBridge API calls
13
+ - **Coordinate-First Design**: Before generating code, the model calculates exact positions, velocities, and trajectories
14
+
15
+ ## 🌟 Key Capabilities
16
+
17
+ ### 1. **Spatial Understanding** 🗺️
18
+ - Deep comprehension of 3D object positions and relationships
19
+ - Precise coordinate calculations (x, y, z positions)
20
+ - Automatic trajectory derivation from physics principles
21
+ - Explicit spatial reasoning in every response
22
+
23
+ ### 2. **Physics Modeling** ⚙️
24
+ - Proficient in classical mechanics, electromagnetism, thermodynamics
25
+ - **API-Driven**: Generates PhysicsBridge API calls directly
26
+ - Supports rigid body collisions, SPH fluids, ODE/PDE solvers
27
+ - Physics principles → API translation in reasoning chain
28
+
29
+ ### 3. **Advanced Problem Solving** 🏆
30
+ - Optimized for CPhO (Chinese Physics Olympiad) level problems
31
+ - Optimized for IMO (International Mathematical Olympiad) level problems
32
+ - Handles complex multi-body systems with constraints
33
+ - Deep reasoning for competition-level problems
34
+
35
+ ## Training Details
36
+
37
+ - **Base Model**: OLMo-3.1-32B-Instruct
38
+ - **Fine-tuning Method**: LoRA (Low-Rank Adaptation)
39
+ - **LoRA Rank**: 256
40
+ - **Training Data**: 3,500+ physics and mathematics problems with detailed reasoning chains
41
+ - **Training Framework**: DeepSpeed ZeRO-3 with BF16 precision
42
+ - **Hardware**: 3x RTX 5090 GPUs
43
+
44
+ ## Model Output Format
45
+
46
+ The model generates responses with **explicit spatial reasoning chains** that enforce coordinate thinking:
47
+
48
+ ```
49
+ <thought>
50
+ 【空间直觉分析】
51
+ - 问题的3D空间结构
52
+ - 物体初始位置 (x₀, y₀, z₀)
53
+ - 物体初始速度 (vₓ, vᵧ, vᵤ)
54
+ - 坐标系建立
55
+
56
+ 【物理原理推导】
57
+ - 适用的物理定律
58
+ - 力的分析
59
+ - 加速度计算
60
+
61
+ 【空间坐标计算】
62
+ - 时间 t 时的位置: (x(t), y(t), z(t))
63
+ - 速度向量: (vₓ(t), vᵧ(t), vᵤ(t))
64
+ - 轨迹方程
65
+
66
+ 【物理→API翻译】
67
+ - PhysicsBridge API 调用
68
+ - 参数映射: 坐标 → API 参数
69
+ - 初始条件设置
70
+ </thought>
71
+
72
+ <code>
73
+ # PhysicsBridge API 调用
74
+ physics = PhysicsBridge()
75
+ physics.create_rigid_body(
76
+ position=(x₀, y₀, z₀),
77
+ velocity=(vₓ, vᵧ, vᵤ),
78
+ mass=m,
79
+ shape='sphere'
80
+ )
81
+ # ... 更多 API 调用
82
+ </code>
83
+ ```
84
+
85
+ ### What Makes This Different
86
+
87
+ ✨ **Forced Spatial Thinking**: The model MUST calculate coordinates before writing code
88
+ ✨ **API-First Architecture**: Physics principles are translated directly to PhysicsBridge API
89
+ ✨ **Explicit Reasoning**: Every step of spatial reasoning is visible in the output
90
+
91
+ ## Usage
92
+
93
+ ```python
94
+ from transformers import AutoModelForCausalLM, AutoTokenizer
95
+
96
+ model_id = "SStarrySSky/Askit-OLMo-32B-Spatial-Thinking-Preview"
97
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
98
+ model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
99
+
100
+ # Example: Physics simulation with spatial reasoning
101
+ prompt = """
102
+ Create a physics simulation for a ball dropped from 10 meters.
103
+ The ball has mass 1kg and initial velocity (0, 0, 0).
104
+ Use PhysicsBridge API to create the simulation.
105
+ """
106
+
107
+ inputs = tokenizer(prompt, return_tensors="pt")
108
+ outputs = model.generate(**inputs, max_length=2048, temperature=0.7)
109
+ response = tokenizer.decode(outputs[0])
110
+
111
+ # Output will include:
112
+ # 1. <thought> section with explicit spatial calculations
113
+ # 2. <code> section with PhysicsBridge API calls
114
+ print(response)
115
+ ```
116
+
117
+ ## Use Cases
118
+
119
+ ### 1. **Physics Education** 🎓
120
+ - Generate interactive animations for teaching physics concepts
121
+ - Explicit spatial reasoning helps students understand 3D motion
122
+ - API-driven code is directly executable in Askit platform
123
+
124
+ ### 2. **Mathematical Visualization** 📊
125
+ - Create visual demonstrations of mathematical problems
126
+ - Coordinate calculations ensure geometric accuracy
127
+ - Perfect for IMO-level problem visualization
128
+
129
+ ### 3. **Research Simulation** 🔬
130
+ - Build physics simulations for academic research
131
+ - Spatial thinking ensures correct coordinate systems
132
+ - API integration with PhysicsBridge for real-time rendering
133
+
134
+ ### 4. **Competitive Problem Solving** 🏆
135
+ - Solve CPhO and IMO level problems with visualization
136
+ - Forced spatial reasoning matches competition requirements
137
+ - Generates production-ready simulation code
138
+
139
+ ## Limitations
140
+
141
+ - Optimized for physics and mathematics domains
142
+ - Requires PhysicsBridge API for full functionality
143
+ - Best performance with detailed problem descriptions
144
+
145
+ ## License
146
+
147
+ GPL-3.0 License
148
+
149
+ ## Citation
150
+
151
+ ```bibtex
152
+ @model{askit2024,
153
+ title={Askit-OLMo-32B-Spatial-Thinking-Preview},
154
+ author={Starry Sky},
155
+ year={2024}
156
+ }
157
+ ```
158
+
159
+ ## Acknowledgments
160
+
161
+ Built on top of OLMo-3.1-32B-Instruct by Allen Institute for AI.