#128 · 主分类： 自然语言处理与文本处理
AnglE
dense-retrieval
embeddings
information-retrieval
llama
llama2
llm
mteb
rag
retrieval-augmented-generation
semantic-similarity
semantic-textual-similarity
sentence-embedding
sentence-embeddings
sentence-vector
sts
stsbenchmark
text-embedding
text-similarity
text-vector
text2vec
使用AnglE训练和推理强大的句子嵌入 | 🔥 在STS和MTEB排行榜上达到SOTA
项目最后更新：03/22/26
GitHub Stars
573
Forks数量
38
贡献者数量
10
许可证
MIT
收录理由
AnglE 是一个轻量级的句子嵌入训练与推理库，句子嵌入正是语义搜索、检索增强生成和文本相似度计算背后的核心向量表示。它支持用几行代码从 BERT 类模型或 LLaMA、Mistral、Qwen 等大语言模型训练嵌入，同时也能作为通用推理框架加载其他基于 Transformer 的嵌入模型。对于需要针对自身领域数据定制向量、而非直接使用通用预训练模型的团队，它的训练功能尤其有价值，内置多种对比损失函数，并支持单卡或多卡训练。该方法发表于 ACL 2024 主会论文，技术细节有据可查，不是黑盒；同时提供了预训练模型，想直接使用的用户也可以快速上手。如果你正在构建检索或相似度功能，并且希望掌控嵌入的训练与服务方式，这个库是一个实用的起点。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。