CoolFace
Modelpublic

XuehangCang/MicroMajor-2B-AIAppTech

sourceHugging Faceapache-2.0updated 5mo agoView on Hugging Face
0likes22downloads
Model Card

MicroMajor-2B-AIAppTech 是针对人工智能应用技术微专业方向打造的垂直领域小模型

以 Qwen/Qwen3.5-2B 为基座,使用专属课程 Q&A 数据集 XuehangCang/MicroMajor-AIAppTech 进行有监督微调,覆盖课程如下:

序号课程名称
1人工智能概述
2Python 程序设计实践
3机器学习基础
4自然语言处理应用与实践
5人工智能实践

评测结果

评测方法:25 道专业题(5 课程 × 5 题),与基座模型 Qwen3.5-2B 进行对比 评分规则:关键词命中率得分(0–6 分)+ 回答完整性得分(0–4 分)= 综合得分(0–10 分)

各课程均分对比

课程MicroMajor-2B-AIAppTechQwen3.5-2B提升幅度
人工智能概述8.407.52+0.88
Python程序设计实践8.637.77+0.86
机器学习基础9.148.46+0.68
自然语言处理应用与实践8.487.99+0.49
人工智能实践8.637.60+1.03
全科综合8.667.87+0.79

多维度对比

维度MicroMajor-2B-AIAppTechQwen3.5-2B
关键词均分(/6)4.663.87
完整性均分(/4)4.004.00
综合均分(/10)8.667.87
平均回答字符数836 字830 字
平均推理耗时11.2 秒11.3 秒

模型信息

属性详情
基座模型Qwen/Qwen3.5-2B
模型架构Qwen3_5ForCausalLM
参数量~2B
隐藏层维度2048
Transformer 层数24
注意力头数8(KV 头数:2)
最大上下文长度262,144 tokens
推理精度bfloat16
语言中文为主

训练详情

数据集

  • —数据集:XuehangCang/MicroMajor-AIAppTech
  • —格式:question / thinking / answer 三字段,助手回复使用 <think>…</think> 包裹推理链,后接最终答案(Chain-of-Thought 风格)

微调方法

  • —方法:LoRA SFT
  • —框架:🤗 Transformers + TRL + PEFT

超参数配置

超参数值
微调方法LoRA
LoRA Rank (r)16
LoRA Alpha32
LoRA Dropout0.05
LoRA 目标模块all-linear(q/k/v/oproj、gate/up/downproj 等全部线性层)
训练轮次3 epochs
单卡 Batch Size2
梯度累积步数8(等效全局 batch = 16)
学习率2e-4
学习率调度Cosine
Warmup 比例5%
权重衰减0.01
优化器AdamW Fused
训练精度bfloat16
最大序列长度2048 tokens
梯度检查点✅ 开启

快速开始

环境依赖

bash
pip install transformers>=4.51.0 torch accelerate

推理示例

python
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_path = "XuehangCang/MicroMajor-2B-AIAppTech"

tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    model_path,
    torch_dtype=torch.bfloat16,
    device_map="auto",
    trust_remote_code=True,
)
model.eval()

question = "请简述 Transformer 架构中 Self-Attention 的计算过程。"
messages = [{"role": "user", "content": question}]

text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True,
)
inputs = tokenizer(text, return_tensors="pt").to(model.device)

with torch.no_grad():
    output_ids = model.generate(
        **inputs,
        max_new_tokens=512,
        temperature=0.7,
        top_p=0.9,
        do_sample=True,
        pad_token_id=tokenizer.eos_token_id,
    )

new_ids = output_ids[0][inputs["input_ids"].shape[-1]:]
print(tokenizer.decode(new_ids, skip_special_tokens=True))

使用 pipeline

python
from transformers import pipeline

pipe = pipeline(
    "text-generation",
    model="XuehangCang/MicroMajor-2B-AIAppTech",
    torch_dtype="bfloat16",
    device_map="auto",
)

messages = [{"role": "user", "content": "什么是 RAG?它在 AI 应用中有什么作用?"}]
result = pipe(messages, max_new_tokens=512, temperature=0.7)
print(result[0]["generated_text"][-1]["content"])

适用场景

  • —🎓 AI 专业课程辅助:为学生提供人工智能应用技术各科目的问答辅导
  • —📚 智能题库与练习:生成专业习题解析、概念解释
  • —🤖 课程智能助手:集成至教学平台,回答学生关于课程知识点的问题
  • —🔬 垂直领域研究:作为教育领域小模型微调方法的参考实现

局限性

  • —本模型专注于人工智能应用技术垂直领域,通用知识覆盖相对有限
  • —评测基于关键词匹配与字符数统计,建议结合人工评估使用
  • —参数量为 2B,复杂推理任务建议配合 Chain-of-Thought 提示词
  • —模型回答可能存在幻觉,请结合教材内容进行核实

引用

如果本模型对您的研究或教学有所帮助,欢迎引用:

bibtex
@misc{MicroMajor-2B-AIAppTech,
  author       = {XuehangCang},
  title        = {MicroMajor-2B-AIAppTech: A Domain-Specific LLM for AI Application Technology Education},
  year         = {2026},
  publisher    = {Hugging Face},
  url          = {https://huggingface.co/XuehangCang/MicroMajor-2B-AIAppTech}
}

许可证

本模型基于 Qwen3.5-2B 微调,遵循 Apache 2.0 License