Modelpublic

OldKingMeister/llama-3-8b-instruct-lmsys-arena-final

sourceHugging Facellama3.1updated 7mo agoView on Hugging Face

0likes9downloads

Model Card

Llama 3 8B Instruct - LMSYS Chatbot Arena LoRA Adapter

This is a LoRA (Low-Rank Adaptation) adapter checkpoint for the meta-llama/Meta-Llama-3-8B-Instruct model, fine-tuned for the LMSYS Chatbot Arena Competition on Kaggle.

Model Description

This model was trained to predict human preferences between two chatbot responses, also known as reward modeling or preference modeling. Given two responses (Response A and Response B), the model outputs a score indicating which response is preferred by humans.

This variant uses the Instruct-tuned version of Llama 3 8B as the base model, which may provide better understanding of instruction-following and conversational contexts.

—Base Model: meta-llama/Meta-Llama-3-8B-Instruct
—Adapter Type: LoRA (Low-Rank Adaptation)
—Task: Sequence Classification (preference prediction)
—Competition: LMSYS Chatbot Arena
—Framework: PEFT (Parameter-Efficient Fine-Tuning)

LoRA Configuration

json

{
  "peft_type": "LORA",
  "r": 16,
  "lora_alpha": 32,
  "lora_dropout": 0.1,
  "target_modules": ["o_proj", "k_proj", "v_proj", "q_proj"],
  "modules_to_save": ["classifier", "score"],
  "task_type": "SEQ_CLS"
}

Usage

Installation

bash

pip install transformers peft torch

Inference

python

from peft import PeftModel
from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

# Load base model
base_model = AutoModelForSequenceClassification.from_pretrained(
    "meta-llama/Meta-Llama-3-8B-Instruct",
    num_labels=1,
    torch_dtype=torch.float16,
    device_map="auto"
)

# Load LoRA adapter
model = PeftModel.from_pretrained(base_model, "OldKingMeister/llama-3-8b-instruct-lmsys-arena-final")
tokenizer = AutoTokenizer.from_pretrained("OldKingMeister/llama-3-8b-instruct-lmsys-arena-final")

# Prepare input - example comparing two responses
text = """Which response is better for the prompt: What is machine learning?

Response A: Machine learning is a subset of AI.

Response B: Machine learning enables systems to learn from experience."""

# Tokenize and predict
inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
inputs = {k: v.to(model.device) for k, v in inputs.items()}

with torch.no_grad():
    outputs = model(**inputs)
    preference_score = outputs.logits.item()

# Score interpretation:
# Positive: Response B is preferred
# Negative: Response A is preferred
print(f"Preference score: {preference_score}")

Training Details

Dataset

—Source: LMSYS Chatbot Arena Conversations
—Task: Binary preference classification (which response is better)
—Format: Conversations with human preference labels

Training Hyperparameters

Parameter	Value
Learning Rate	2e-4
Batch Size	4
Gradient Accumulation Steps	4
Epochs	6
Max Sequence Length	512
LoRA Rank (r)	16
LoRA Alpha	32
LoRA Dropout	0.1

Hardware

—GPU: NVIDIA A100 (40GB)
—Training Time: ~8 hours
—Mixed Precision: fp16

Model Architecture

—Base Parameters: 8B (frozen)
—Trainable Parameters: ~54M (LoRA adapters + classification head)
—Total Checkpoint Size: ~82MB

Citation

bibtex

@misc{lmsys-arena-2024,
  title={LMSYS Chatbot Arena Competition},
  howpublished={https://www.kaggle.com/competitions/lmsys-chatbot-arena},
  year={2024}
}

@article{llama3-2024,
  title={The Llama 3 Herd of Models},
  author={{Meta AI}},
  year={2024}
}

License

Llama 3.1 License (see base model for details)