Atlas

E-AI Project

Llama-3.1-6B β€” 35% Compressed from Llama-3.1-8B (English)

Built with Llama. This repository is part of the Efficient and Robust AI System (E-AI) Project by Vincent-Daniel Yun. This model is a compressed edition of meta-llama/Llama-3.1-8B with 11 of 32 transformer layers removed (21 layers remain, β‰ˆ5.63B parameters), recovered training-free. It is a base (non–chat-tuned) model.

πŸ”— Project: https://www.worldwidedaniel.com/eai-project πŸ“… Release: 2026-06-28 Β· Version: V1

βš–οΈ License: governed by the Llama 3.1 Community License Agreement (https://huggingface.co/meta-llama/Llama-3.1-8B/blob/main/LICENSE), and subject to the Llama 3.1 Acceptable Use Policy (https://www.llama.com/llama3_1/use-policy). By using this model you agree to those terms. "Llama" is a trademark of Meta Platforms, Inc. Built with Llama.

⚠️ Language: English-focused. Use as a discrimination / classification engine (classification, reading comprehension, domain QA, judging, safety screening) β€” open-ended long-form generation is degraded by compression (see PPL). For factual questions use retrieval (RAG).

About E-AI

Modern AI is powerful but heavy. State-of-the-art models are enormous and their inference is slow. The E-AI (Efficient-AI) project builds compact yet capable AI β€” making every model lightweight and fast, and keeping multi-agent teams reliable even when some members fail β€” so AI can assist people in urgent, high-stakes moments.

Method

The pruning method and the training-free recovery method are proprietary, undisclosed methods created by Vincent-Daniel Yun and are not released. Only the resulting model is shared.

Results (measured, full test sets)

PPL on 2048-token context (↓ better); downstream & MMLU are 0-shot via lm-eval-harness (↑ better).

Metric Llama-3.1-8B (base) This model (35%)
PPL Β· WikiText2 ↓ 6.24 73.74
PPL Β· C4 ↓ 8.68 65.51
ARC-c ↑ 0.5350 0.3848
ARC-e ↑ 0.8114 0.5661
BoolQ ↑ 0.8208 0.7086
COPA ↑ 0.8700 0.7200
HellaSwag ↑ 0.7889 0.5605
OpenBookQA ↑ 0.4480 0.3200
RACE ↑ 0.3914 0.3435
RTE ↑ 0.6931 0.7184
WinoGrande ↑ 0.7356 0.6701
Avg. downstream (9) ↑ 0.6771 0.5547
MMLU ↑ 0.6360 0.6223

Task-suitability (vs base Llama-3.1-8B)

Where this compressed model stays close to the base on discrimination tasks (full test sets):

Task Llama-3.1-8B (base) This model (35%)
Topic classification (AG News) 0.564 0.584
LLM-as-judge (RewardBench) 0.660 0.667
SafetyBench (MCQ) 0.743 0.629
MultiRC 0.572 0.572
WiC 0.511 0.500
MRPC 0.674 0.684
CB (NLI) 0.643 0.804
SST-2 0.775 0.802
MedQA 0.598 0.596
MedMCQA 0.564 0.559
PubMedQA 0.758 0.718
Belebele-en 0.788 0.772
Belebele-ko 0.662 0.642
XNLI-zh 0.395 0.366
ToxiGen 0.432 0.431
TruthfulQA 0.452 0.506

(Discrimination is largely preserved; generation/PPL degrades with compression. Yes/No safety-F1 benchmarks are omitted as the non-chat base is poorly calibrated for them.)

Efficiency (measured, fp16, single GPU)

Llama-3.1-8B (base) This model (35%)
Layers 32 21
Parameters 8.03B 5.63B
Peak inference memory (fp16) 19.31 GB 14.14 GB (βˆ’27%)
Forward latency (fp16) 1293 ms 852 ms (βˆ’34%)

Usage β€” Transformers

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
m = AutoModelForCausalLM.from_pretrained("daniel-eai/Llama-3.1-6B-35pct-Compressed-8B-EN-V1", trust_remote_code=True, dtype=torch.float16, device_map="cuda")
tok = AutoTokenizer.from_pretrained("daniel-eai/Llama-3.1-6B-35pct-Compressed-8B-EN-V1", trust_remote_code=True)
ids = tok("The capital of France is", return_tensors="pt").to("cuda")
print(tok.decode(m.generate(**ids, max_new_tokens=20)[0]))

trust_remote_code=True is required (custom decoder layer in modeling_llama_recovered.py).

License & Acknowledgements

This model is a derivative of Llama 3.1 and is distributed under the Llama 3.1 Community License. Built with Llama. Thanks to Prof. Sai Praneeth Karimireddy (USC) and Prof. Sunwoo Lee (Inha University) for guidance, and to Meta for releasing Llama 3.1.

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

Model tree for daniel-eai/Llama-3.1-6B-35pct-Compressed-8B-EN-V1

Finetuned
(1435)
this model