Tip
这个仓库包含了三个深度学习相关的实验项目实现,涵盖神经网络基础实现、图像分割模型优化和微调微型大语言模型。
graph TD
A[深度学习项目] --> B[实验一: 多层神经网络的NumPy实现]
A --> C[实验二: TinySeg图像分割模型对比与优化]
A --> D[实验三: 基于MiniMind的微型大语言模型研究]
B --> E[model.py - 神经网络模型实现]
B --> F[train.py - 训练脚本]
B --> G[utils.py - 工具函数]
B --> H[可视化结果]
C --> I[models/ - 分割模型实现]
C --> J[dataloader.py - 数据加载]
C --> K[train.py - 训练脚本]
C --> L[evaluate.py - 评估脚本]
D --> M[minimind/ - 微型LLM实现]
D --> N[train_*.py - 各种训练脚本]
D --> O[eval_model.py - 评估脚本]
这个实验展示了如何使用NumPy从零开始实现一个多层神经网络,用于解决MNIST手写数字分类问题。
本实验对比研究了PSPNet、DeepLabV3和CCNet三种图像分割模型,通过优化训练策略和模型结构提升分割性能。
本项目从零开始实现了一个仅25.8M参数的微型大语言模型MiniMind,包含预训练、微调、LoRA、DPO强化学习和模型蒸馏等完整流程。
实验一
- 🔧 纯NumPy实现,不依赖深度学习框架
- 📊 完整的可视化支持(训练过程、混淆矩阵、错误样本分析)
- 🎯 模块化设计,代码结构清晰
实验二
- 🔍 三种主流分割模型的对比研究
- ⚡ 训练策略和模型结构的优化方案
- 📈 详细的性能评估和分析
实验三
- 💡 从零实现25.8M参数的微型LLM
- 🚀 包含预训练、SFT、LoRA、DPO全流程
- ⏱️ 低成本训练(3元+2小时)
- 🧩 兼容主流框架(transformers/trl/peft)
- Python 3.9+
- NumPy
- Matplotlib
- Seaborn
- Scikit-learn
- torch
- transformers
- ...
- 克隆仓库
git clone git@github.com:Alleyf/DeepLearning.git
cd DeepLearning- 安装依赖
# 方法1:直接从requirements.txt安装所有依赖
pip install -r requirements.txt- 运行实验
cd "experiment_one_numpy implementation of multi-layer neural network"
python train.py本项目基于 APACHE 2.0 许可证开源。
欢迎提交 Issue 和 Pull Request 来帮助改进项目!