
RAG-Retrieval
统一微调 RAG 检索模型,让召回更精准
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 1137
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 3.8(README 指定用 conda 创建同名环境)
- 已安装 conda,可创建并激活虚拟环境
- 与本地 CUDA 版本兼容的 PyTorch(README 建议手动安装)
- pip(用于安装训练依赖与 rag-retrieval 推理库)
安装与启动步骤
-
1创建并激活环境
用 conda 建立 Python 3.8 的独立环境,避免与其他项目依赖冲突;创建后必须在同一终端激活再装包。
conda create -n rag-retrieval python=3.8 && conda activate rag-retrieval -
2获取项目代码
训练相关脚本和 requirements.txt 都在仓库内,需先克隆到本地并进入项目根目录。
git clone https://github.com/NovaSearch-Team/RAG-Retrieval.git cd RAG-Retrieval -
3安装训练依赖
在项目根目录执行;README 建议先手动安装与本地 CUDA 兼容的 torch,再装依赖,避免自动安装的 torch 版本不匹配。
pip install -r requirements.txt -
4安装推理库
仅做 reranker 预测时使用,与训练依赖分开安装,同一环境内执行 pip 安装即可。
pip install rag-retrieval
如何确认成功
执行 pip show rag-retrieval 能显示已安装的包信息,即说明预测库安装成功;README 未给出训练成功与否的判定方式。
常见问题
Q:只想做 reranker 预测,需要装训练依赖吗?
A:不需要。README 区分了两条路径:训练用 pip install -r requirements.txt,预测用 pip install rag-retrieval。
Q:torch 和本地 CUDA 不兼容怎么办?
A:按 README 建议,在装 requirements.txt 或 rag-retrieval 之前,先手动 pip 安装与本地 CUDA 匹配的 torch 版本,再执行后续安装。
Q:这个框架能微调哪些检索模型?
A:支持任意开源 RAG 检索模型微调,包括 embedding 模型(bert 系、llm 系)、延迟交互模型 colbert 以及 reranker 模型(bert 系、llm 系)。
Q:支持把大模型蒸馏到小模型吗?
A:支持。README 说明可对 embedding 模型和 reranker 模型做蒸馏,从较大模型蒸馏到 0.5b LLM 或 bert-base 等小模型。
注意事项
- README 只提供了安装方式,未给出训练、推理、蒸馏的具体命令与参数,实际使用需自行阅读仓库内脚本与配置。
- 训练与预测是两条独立安装路径,建议按用途选择,避免在同一环境中混装造成依赖冲突。
- README 明确建议先手动安装兼容 CUDA 的 torch,再执行依赖安装,否则可能出现 torch 与本地 CUDA 不匹配。
- README 中的仓库链接与项目主页给出的 GitHub 地址不一致,克隆时请以实际可访问的仓库为准。
核心亮点
- 统一覆盖 Embedding、ColBERT、ReRanker 三类检索模型,一套流程搞定微调
- 提供高效训练策略(如负采样、蒸馏),节省调参时间
- 接口设计清晰,易于集成到现有 RAG 流水线
不足之处
- 项目仍处于成长阶段,文档和社区生态有待完善
- 对新手不够友好,需要一定 RAG 和微调基础
适用场景
- 领域知识库检索优化,提升专业问答准确率
- 企业级 RAG 系统上线前的检索模型定制训练
- 研究对比不同检索模型微调效果的实验平台
替代项目
FlagEmbedding、sentence-transformers、Cohere Reranker
上一篇:GAN-XML-Fixer
下一篇:pylate
同类项目推荐
onebitllms
开源
小显存也能训练1.58bit三值大模型
Lightweight toolkit package to train and fine-tune 1.58bit Language models
RAGEN
开源
多轮智能体强化学习训练与推理退化诊断
Agent RL framework for LLM agents: multi-turn reinforcement learning with StarPO and···
DreamBooth
开源
用几张照片,让AI学会画你的专属对象
This repository contains the implementation of DreamBooth with Stable Diffusion. You···
plexe
开源
一句话描述,自动生成机器学习模型,快速验证想法
✨ Build a machine learning model from a prompt
gflownet
开源
按奖励比例采样多样化候选解的生成模型框架
Generative Flow Networks - GFlowNet
deep-delta-learning
开源
学增量,少训练,模型升级快人一步
Official Project Page for Deep Delta Learning (https://arxiv.org/abs/2601.00417)
Alpaca-CoT
开源
统一接口微调大模型,轻松搞定数据、模型与方法
We unified the interfaces of instruction-tuning data (e.g., CoT data), multiple LLMs···
simple-diffusion
开源
用最简代码看懂扩散模型,快速跑通图像生成
A minimal implementation of a denoising diffusion model in PyTorch.
Awesome-LLM-On-Policy-Distillati···
开源
一网打尽LLM在线策略蒸馏论文,快速跟上研究前沿
A curated collection of papers and resources on On-Policy Distillation for Large Lan···