EXAONE-4.0-1.2B-medical-medium

🎯 MEDICAL-optimized | 📦 Medium pruning | ⚡ 18% weights pruned

This model is a moderately pruned version of LGAI-EXAONE/EXAONE-4.0-1.2B, specialized for MEDICAL tasks using activation-aware weight pruning (Wanda-style).

✨ Key Features

• Specialization: Optimized for Medical tasks
• Pruning Method: Wanda-style (|W| × |activation|) importance scoring
• Size Reduction: 18% weights pruned
• Use Case: Balanced trade-off between size and accuracy

📊 Performance Comparison

Category	Original	Pruned	Change
Python	20.0%	20.0%	→
Html	6.7%	0.0%	↓ 6.7%
Trivia	86.7%	93.3%	↑ 6.7%
Math	60.0%	66.7%	↑ 6.7%
Reasoning	N/A	N/A
Medical	93.3%	80.0% ⭐	↓ 13.3%
Linux	93.3%	80.0%	↓ 13.3%
Writing	46.7%	46.7%	→

Average: 58.1% → 55.2% (-2.9%)

Medical Retention: 85.7% of original performance

🚀 Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("CompactAI/EXAONE-4.0-1.2B-medical-medium")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/EXAONE-4.0-1.2B-medical-medium")

# Example usage
inputs = tokenizer("Your prompt here", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

📋 Technical Details

Property	Value
Base Model	LGAI-EXAONE/EXAONE-4.0-1.2B
Specialization	Medical
Prune Mode	Medium
Pruning Method	Activation-based weight pruning (Wanda)
Weight Reduction	18% weights pruned

🔗 Related Models

This model is part of the EXAONE-4.0-1.2B pruned model collection. Other variants:

• Extra-light (minimal pruning)
• Light
• Medium-light
• Medium
• Medium-heavy
• Heavy
• Extra-heavy (maximum compression)

📜 License

This model inherits the license from the base model LGAI-EXAONE/EXAONE-4.0-1.2B.

---

Generated by ZANNPS [Zeto Automatic Neural Network Pruning System]