CoolFace
Modelpublic

tony-shieh/Qwen2.5-14B-LoRA-NYCU-DL-HW2

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
0likes
Model Card

Qwen2.5-14B-LoRA-NYCU-DL-HW2

Model Description

This model is a Supervised Task Finetuned (SFT) version of Qwen2.5-14B-Instruct, specifically trained with reasoning information (Chain-of-Thought) to tackle complex multiple-choice questions.

It was developed as part of the Deep Learning HW2 coursework at National Yang Ming Chiao Tung University (NYCU).

  • —Developed by: 謝宗穎(Zong-Ying Shieh) 314706019
  • —Base Model: unsloth/Qwen2.5-14B-Instruct-bnb-4bit
  • —Task: Multiple-Choice Question Answering & Logical Reasoning
  • —Language(s): Traditional Chinese (zh-TW), English
  • —License: Apache 2.0

Performance

  • —Kaggle Public Leaderboard Score: 0.72340
  • —Evaluation Method: The reported score is achieved by coupling this SFT model with a highly optimized Test-Time Compute inference strategy: 3-Pass Self-Consistency (SC) combined with Softmax Logits Soft-Voting.

How to Use (Inference)

To achieve the maximum performance, it is highly recommended to use the Logits Extraction method rather than standard text generation. Below is a basic snippet to load the model using unsloth:

python
from unsloth import FastLanguageModel
import torch

model, tokenizer = FastLanguageModel.from_pretrained(
    model_name = "tony-shieh/Qwen2.5-14B-LoRA-NYCU-DL-HW2", 
    max_seq_length = 4096,
    dtype = None,
    load_in_4bit = True 
)
FastLanguageModel.for_inference(model)

System Prompt used during training

SYSTEM_PROMPT = "你是一個專業且中立的選擇題解題專家。請針對題目進行嚴密的邏輯推理,客觀分析每個選項,最後明確給出你的答案。"

Training Details

Hardware & Environment

  • —Hardware: 1x NVIDIA GeForce RTX 5090 (32GB VRAM)
  • —raining Framework: unsloth, trl, transformers
  • —Memory Optimization: Overcame Out-of-Memory (OOM) constraints on a single 32GB GPU by utilizing PyTorch's expandablesegments:True, drastically reducing the perdevicetrainbatchsize to 1, and compensating with gradientaccumulation_steps=16 to maintain a stable effective batch size.

Hyperparameters

LoRA Rank (r): 32

LoRA Alpha: 64

Target Modules: qproj, kproj, vproj, oproj, gateproj, upproj, down_proj

Epochs: 2 (To prevent catastrophic forgetting and overfitting on the public LB)

Learning Rate: 1e-4

Optimizer: adamw_8bit

Inference Strategy

The true potential of this model is unlocked during the inference phase. Instead of relying on vulnerable Regex text parsing, the final submission utilizes:

  1. 1.Dynamic Chain-of-Thought (do_sample=True, temp=0.6): Forcing the model to explore 3 distinct reasoning paths for the same question.
  1. 1.Logits Soft-Voting: Extracting the raw neural network logits for tokens A, B, C, and D at the final layer, normalizing them via Softmax into confidence probabilities, and accumulating these scores across the 3 iterations. This effectively mitigates the 14B model's "hallucination" and locks in the most logically sound answer.