Qwen3-4B-trivia-medium

🎯 TRIVIA-optimized | πŸ“¦ Medium pruning | ⚑ 15% weights pruned

This model is a moderately pruned version of Qwen/Qwen3-4B, specialized for TRIVIA tasks using activation-aware weight pruning (Wanda-style).

✨ Key Features

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

πŸ“Š Performance Comparison

Category Original Pruned Change
Python 20.0% 80.0% ↑ 60.0%
Html 0.0% 0.0% β†’
Trivia 100.0% 100.0% ⭐ β†’
Math 100.0% 100.0% β†’
Reasoning N/A N/A
Medical 100.0% 100.0% β†’
Linux 100.0% 100.0% β†’
Writing 93.3% 93.3% β†’

Average: 73.3% β†’ 81.9% (+8.6%)

Trivia Retention: 100.0% of original performance

Comparison Graph

πŸš€ Quick Start

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("CompactAI/Qwen3-4B-trivia-medium")
tokenizer = AutoTokenizer.from_pretrained("CompactAI/Qwen3-4B-trivia-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 Qwen/Qwen3-4B
Specialization Trivia
Prune Mode Medium
Pruning Method Activation-based weight pruning (Wanda)
Weight Reduction 15% weights pruned

πŸ”— Related Models

This model is part of the Qwen3-4B 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 Qwen/Qwen3-4B.


Generated by ZANNPS [Zeto Automatic Neural Network Pruning System]

Downloads last month
-
Safetensors
Model size
4B params
Tensor type
F16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for CompactAI/Qwen3-4B-trivia-medium

Base model

Qwen/Qwen3-4B-Base
Finetuned
Qwen/Qwen3-4B
Finetuned
(482)
this model

Collection including CompactAI/Qwen3-4B-trivia-medium