updated readme

README.md

@@ -23,8 +23,8 @@ datasets:
 
 # DAGGER-4B-SFT-GRPO
 
-<a href="https://arxiv.org/abs/XXXX.XXXXX" target="_blank">
-    <img alt="arXiv" src="https://img.shields.io/badge/arXiv-XXXX.XXXXX-b31b1b" style="display: inline-block; vertical-align: middle;"/>
+<a href="https://arxiv.org/abs/2601.06853" target="_blank">
+    <img alt="arXiv" src="https://img.shields.io/badge/arXiv-2601.06853-b31b1b" style="display: inline-block; vertical-align: middle;"/>
 </a>
 <a href="https://github.com/dipta007/dagger" target="_blank">
     <img alt="GitHub" src="https://img.shields.io/badge/GitHub-Code-black" style="display: inline-block; vertical-align: middle;"/>
@@ -81,18 +81,66 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
-question = "মিনার কাছে ১০০টি কলম আছে। প্রতিটি কলমের দাম ৫ টাকা।"
+USER_PROMPT_TEMPLATE = """You are an expert Bengali Math Reasoner. Your task is to solve mathematical problems by constructing a "Computational Graph".
+
+### Graph Rules:
+- `id`: Unique identifier (e.g., "n1", "n2").
+- `val`: The raw number extracted from text (for input nodes).
+- `op`: The operation (`add`, `sub`, `mul`, `div`, `round`, `sqrt`, `floor`, `sum`, `mean`, `ratio_split`). Use `const` for input numbers.
+- `args`: List of input node IDs.
+- `distractor`: Boolean (`true` / `false`). Set to `true` if the node is NOT used in the final calculation path.
+- `label`: Label for the node.
+
+### Available Operations:
+- Input: `const` (Use this for all numbers found in text or constants).
+- Arithmetic: `add`, `sub`, `mul`, `div`, `abs` (absolute difference).
+- Logic/Stats: `sum`, `mean`, `min` (minimum), `max` (maximum).
+- Rounding: `round` (nearest int), `floor` (round down), `ceil` (round up).
+- Advanced: `sqrt`, `pow`, `mod` (remainder), `gcd`, `lcm`.
+- Output: `identity` ("final_result" points to the answer node)
+
+Only output a JSON graph representing the solution, nothing else. Nodes must be topologically sorted, and there must be exactly one "final_result" node that represents the final answer. One example is provided below.
+
+### Example:
+Question:
+মিনার কাছে ১২২১৯৫ টা কলম আছে। রাজুর কাছে ২৫০৮৪ টা কলম আছে। মিনা রাজুর কাছে ১১২৬ টি কলম চাইল। রাজু ১০০০ টি কলম দিতে রাজি হল, কিন্তু পরে আর দিলেনা। প্রতিটি কলমের দাম ৪৫.৬ টাকা। মিনা যদি কলমগুলো বিক্রি করতে চায়, সে কত টাকা পাবে?
+
+Output:
+```json
+{{
+  "nodes": [
+    {{"id": "n1", "op": "const", "val": 122195, "distractor": false, "label": "মিনার কলম"}},
+    {{"id": "n2", "op": "const", "val": 25084, "distractor": true, "label": "রাজুর কলম"}},
+    {{"id": "n3", "op": "const", "val": 1126, "distractor": true, "label": "মিনা রাজুর কাছে চাইল"}},
+    {{"id": "n4", "op": "const", "val": 1000, "distractor": true, "label": "রাজু দিতে রাজি হল"}},
+    {{"id": "n5", "op": "const", "val": 45.6, "distractor": false, "label": "প্রতিটি কলমের দাম"}},
+    {{"id": "total_money", "op": "mul", "args": ["n1", "n5"], "distractor": false, "label": "মিনার মোট টাকা"}},
+    {{"id": "final_result", "op": "identity", "args": ["total_money"], "distractor": false, "label": "চূড়ান্ত উত্তর"}}
+  ]
+}}```
+
+### Your Task:
+
+Question:
+{question}
+
+Output:
+"""
+
+question = "রজারের 5টি টেনিস বল আছে। সে আরও 2 ক্যান টেনিস বল কিনেছে। প্রতিটি ক্যানে 3টি করে টেনিস বল আছে। তার কাছে এখন কতগুলি টেনিস বল আছে?"
+prompt = USER_PROMPT_TEMPLATE.format(question=question)
 
 messages = [
-    {"role": "system", "content": "You are an expert Bangla Math Reasoner. Solve by constructing a Computational Graph."},
-    {"role": "user", "content": question}
+  {"role": "user", "content": prompt}
 ]
 
 text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-inputs = tokenizer([text], return_tensors="pt").to(model.device)
+inputs = tokenizer(text, return_tensors="pt").to(model.device)
 
-outputs = model.generate(**inputs, max_new_tokens=1024)
+# Generate
+outputs = model.generate(**inputs, max_new_tokens=1024, temperature=0.7, top_p=0.8)
 response = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
+
 print(response)
 ```
 
@@ -132,5 +180,13 @@ Same as 12B variant:
 ## Citation
 
 ```bibtex
-will be updated
+@misc{nazi2026dagdaggerdistractorawaregraphgeneration,
+      title={{\dag}DAGGER: Distractor-Aware Graph Generation for Executable Reasoning in Math Problems}, 
+      author={Zabir Al Nazi and Shubhashis Roy Dipta and Sudipta Kar},
+      year={2026},
+      eprint={2601.06853},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2601.06853}, 
+}
 ```