Update pipeline tag and add library metadata

README.md

@@ -1,17 +1,22 @@
 ---
-license: mit
+base_model:
+- Qwen/Qwen3-4B-Instruct-2507
 datasets:
 - yaful/MAGE
 language:
 - en
-base_model:
-- Qwen/Qwen3-4B-Instruct-2507
-pipeline_tag: reinforcement-learning
+license: mit
+pipeline_tag: text-generation
+library_name: peft
+arxiv: 2602.08934
 ---
+
 # StealthRL LoRA Adapter for Qwen3-4B-Instruct (PEFT)
 
 This repository hosts a **LoRA (Low-Rank Adaptation) adapter** for the base model  
-**Qwen/Qwen3-4B-Instruct-2507**, trained using the **StealthRL** methodology.
+**Qwen/Qwen3-4B-Instruct-2507**, presented in the paper [StealthRL: Reinforcement Learning Paraphrase Attacks for Multi-Detector Evasion of AI-Text Detectors](https://huggingface.co/papers/2602.08934).
+
+The authors of the paper are Suraj Ranganath and Atharv Ramesh.
 
 It is an **adapter-only** release (PEFT). The full base model is not included and must be downloaded separately from Hugging Face.
 
@@ -21,14 +26,14 @@ It is an **adapter-only** release (PEFT). The full base model is not included an
 
 **StealthRL** is a reinforcement learning framework for generating **adversarial paraphrases** that evade **multiple AI-text detectors** while preserving semantics.
 
-From the paper:
+Key contributions from the paper:
 - StealthRL trains a **paraphrase policy** against a **multi-detector ensemble**
 - Uses **Group Relative Policy Optimization (GRPO)** with **LoRA adapters** on **Qwen3-4B**
 - Optimizes a **composite reward** that balances **detector evasion** with **semantic preservation**
 - Evaluates transfer to a **held-out detector family**, suggesting shared vulnerabilities rather than detector-specific brittleness
 
-Paper: https://arxiv.org/abs/2602.08934  
-Code: https://github.com/suraj-ranganath/StealthRL
+Paper: [https://arxiv.org/abs/2602.08934](https://arxiv.org/abs/2602.08934)  
+Code: [https://github.com/suraj-ranganath/StealthRL](https://github.com/suraj-ranganath/StealthRL)
 
 ---
 
@@ -57,7 +62,7 @@ from peft import PeftModel
 import torch
 
 base_model = "Qwen/Qwen3-4B-Instruct-2507"
-adapter_repo = "YOUR_HF_USERNAME/YOUR_ADAPTER_REPO"
+adapter_repo = "suraj-ranganath/StealthRL-Qwen3-4B-LORA" # Update with actual repo path if different
 
 tokenizer = AutoTokenizer.from_pretrained(base_model, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(
@@ -107,8 +112,8 @@ print(tokenizer.decode(out[0], skip_special_tokens=True))
 
 ## Associated Paper and Code
 
-- **Paper (arXiv)**: https://arxiv.org/abs/2602.08934  
-- **GitHub Repository**: https://github.com/suraj-ranganath/StealthRL
+- **Paper (arXiv)**: [https://arxiv.org/abs/2602.08934](https://arxiv.org/abs/2602.08934)  
+- **GitHub Repository**: [https://github.com/suraj-ranganath/StealthRL](https://github.com/suraj-ranganath/StealthRL)
 
 ---