Candle 运行 XLM-RoBERTa 实战：Fill-Mask、Reranker 与文本分类三大任务指南-北京尧图网络

人工智能大模型机器学习深度学习本地部署模型推理服务【免费下载链接】candleMinimalist ML framework for Rust项目地址https://gitcode.com/GitHub_Trending/ca/candle点击查看免费下载本文基于 Candle 开源仓库中的 xlm-roberta 示例与对应源码完整讲解如何在 Rust 生态中加载 XLM-RoBERTa 系列模型并一次掌握三项核心能力用fill-mask任务让模型补全被mask遮蔽的词、用reranker任务对给定查询做文档相关性重排Reranking以及用text-classification任务输出文本的正式度formality分类分数。读完本文你将能直接复现示例的全部命令并理解其背后的模型加载、分词、前向计算与打分流程为后续接入自己的检索重排或文本分类服务打下基础。XLM-RoBERTa 在 Candle 中的应用场景XLM-RoBERTaXLMRoberta是面向多语言的 Transformer 编码器模型在 Candle 的示例中被特别强调用于reranking重排给定一条查询query和一批候选文档documents模型为每个「查询-文档」对打出一个相关性分数再按分数从高到低对文档排序从而精排召回结果。除了重排同一模型家族还支持掩码语言建模fill-mask与句子级分类。本示例在仓库中的位置是 candle-examples/examples/xlm-roberta/包含一份 Readme.md 和完整可运行的 main.rs模型结构实现位于 candle-transformers/src/models/xlm_roberta.rs。快速上手三种任务的运行命令示例通过--task参数切换三种任务核心命令如下建议加--release获得更快的推理速度Fill Mask掩码补全cargo run --example xlm-roberta --release -- --task fill-mask --model xlm-roberta-base预期输出示例内置的三句测试文本Sentence: 0 : Hello Im a fashion model. Sentence: 1 : Im a little boy. Sentence: 2 : Im living in berlin.Reranker查询-文档重排cargo run --example xlm-roberta --release -- --task reranker --model bge-reranker-base预期输出示例内置查询what is panda?对 5 条文档的重排结果Ranking Results: -------------------------------------------------------------------------------- Rank #4 | Score: 0.0001 | South Korea is a country in East Asia. Rank #5 | Score: 0.0000 | There are forests in the mountains. Rank #2 | Score: 0.7314 | Pandas look like bears. Rank #3 | Score: 0.6948 | There are some animals with black and white fur. Rank #1 | Score: 0.9990 | The giant panda (Ailuropoda melanoleuca), sometimes called a panda bear or simply panda, is a bear species endemic to China. --------------------------------------------------------------------------------可以看到描述大熊猫的完整词条拿到 0.9990 的最高分排在 Rank #1而「韩国是东亚国家」这类无关文档只有 0.0001 分直观展示了重排模型的相关性区分能力。Text Classification文本分类 / 正式度cargo run --example xlm-roberta -- --task text-classification --model xlmr-formality-classifier预期输出对 4 条文本的 formal/informal 概率Formality Scores: Text 1: I like you. I love you formal: 0.9933 informal: 0.0067 Text 2: Hey, whats up? formal: 0.8812 informal: 0.1188 Text 3: Siema, co porabiasz? formal: 0.9358 informal: 0.0642 Text 4: I feel deep regret and sadness about the situation in international politics. formal: 0.9987 informal: 0.0013命令行参数详解示例使用clap定义命令行参数完整参数见 main.rs 的 Args 结构参数默认值说明--taskreranker任务类型fill-mask/reranker/text-classification--modelbge-reranker-base模型选择见下方枚举--model-id无按任务自动映射手动指定 Hugging Face 模型标识如FacebookAI/xlm-roberta-base--revisionmain拉取模型的版本分支--cpu关强制在 CPU 上运行不加则优先使用 GPUCUDA/Metal--tracing关开启 tracing生成trace-timestamp.json供性能分析--tokenizer-file无自动下载本地分词器文件路径替代自动下载的tokenizer.json--config-file无自动下载本地配置文件路径替代自动下载的config.json--weight-files无自动下载本地权重文件路径替代自动下载的权重--prompt无指定计算 embedding 的 prompt示例中暂未使用该分支--model参数支持的模型枚举定义在 main.rs共 6 个取值bge-reranker-base→ 对应BAAI/bge-reranker-basebge-reranker-large→ 对应BAAI/bge-reranker-largebge-reranker-base-v2→ 对应BAAI/bge-reranker-base-v2-m3xlm-roberta-base→ 对应FacebookAI/xlm-roberta-basexlm-roberta-large→ 对应FacebookAI/xlm-roberta-largexlmr-formality-classifier→ 对应s-nlp/xlmr_formality_classifier任务与模型之间存在强约束fill-mask 只接受xlm-roberta-base/largereranker 只接受三个 BGE 模型text-classification 只接受 formality 分类器。若组合不匹配程序会直接bail!报错见 main.rs例如用 BGE 模型跑 fill-mask 会提示 BGE models are not supported for fill-mask task。模型、分词器与权重的加载流程无论执行哪个任务程序都遵循同一套资源加载逻辑值得单独拆解解析 model_id未指定--model-id时根据--task与--model的合法组合映射出默认模型标识从 Hub 拉取资源通过Api::new()与api.model(model_id).with_revision(args.revision)建立仓库句柄依次获取tokenizer.json分词器、config.json模型配置与权重文件权重格式自适应优先尝试model.safetensors失败则回退到pytorch_model.bin见 main.rs。safetensors 走VarBuilder::from_mmaped_safetensors做内存映射零拷贝加载并指定DType::F16pth 文件则走VarBuilder::from_pth同样统一加载为 F16见 main.rs解析配置将config.json反序列化为candle_transformers::models::xlm_roberta::Config配置分词器启用 padding策略为BatchLongest按 batch 内最长序列补齐pad_token_id取自模型配置并显式关闭截断with_truncation(None)见 main.rs选择设备调用candle_examples::device(args.cpu)定义在 candle-examples/src/lib.rsCPU 模式直接落到 CPU否则自动检测并优先 CUDA/Metal。任务一Fill-Mask 掩码补全该任务的目标是给定包含mask占位符的句子预测最可能填充该位置的词。示例内置三句测试文本let prompt vec![ Hello Im a mask model..to_string(), Im a mask boy..to_string(), Im mask in berlin..to_string(), ];流程如下对应 main.rs构建XLMRobertaForMaskedLM模型用tokenize_batch把句子批编码为input_ids用get_attention_mask取出 attention masktoken_type_ids则直接以零张量填充XLM-R 编码器输入中该维度保留占位调用model.forward得到形状为[batch, seq_len, vocab_size]的 logits在词表维度dim 2上argmax取出每个位置最可能的 token id再decode_batch解码为文本输出。输出时按句子逐条打印Sentence: N : 解码结果。由于掩码处的预测来自整个词表的 argmax解码结果即模型认为最合适的补全词。模型实现上XLMRobertaForMaskedLM在 RoBERTa 主干之上叠加了XLMRobertaLMHead先经过 dense 线性层与 GELU 激活、LayerNorm再与共享的词嵌入矩阵做广播矩阵乘broadcast_matmul(shared_embeddings)得到词表 logits见 xlm_roberta.rs。任务二Reranker 文档重排Reranker 是本文档强调的核心场景。示例内置查询what is panda?与 5 条候选文档核心逻辑见 main.rs将查询与每条文档两两配对构造(query, doc)二元组列表用TokenizeInput::Pairs进行成对编码encode_batch对句对编码时会自然加入 XLM-R 的句对分隔 token得到input_ids与attention_mask构建XLMRobertaForSequenceClassification::new(1, config, vb)——num_labels 为 1即输出单个相关性 logit前向得到[batch, 1]的 logits通过candle_nn::ops::sigmoid归一化到 (0,1) 区间作为相关性分数用arg_sort_last_dim(false)按分数降序得到排名索引逐条打印Rank #N | Score: x.xxxx | 文档内容。从输出可见分数分布拉开明显差距0.9990 到 0.0000这正是交叉编码器cross-encoder式重排模型的特点查询与文档拼接后整体编码相关性判断更精细但代价是每条文档都要一次前向。任务三Text-Classification 文本分类该任务以「正式度分类」为例判断一句话属于formal还是informal。示例内置 4 条文本包括英语、波兰语等不同语言输入体现 XLM-R 的多语言能力。流程见 main.rs构建XLMRobertaForSequenceClassification::new(2, config, vb)num_labels 为 2formal / informal 两个类别对句子批做单句编码前向得到[batch, 2]的 logits在类别维上用candle_nn::ops::softmax归一化得到两个类别的概率按formal probs[0]、informal probs[1]格式化输出。分类头XLMRobertaClassificationHead的实现见 xlm_roberta.rs遵循 XLM-R 原始实现取序列[CLS]位置的隐状态get_on_dim(1, 0)依次经过 dense 层、tanh 激活、out_proj 输出层得到类别 logits。模型结构源码解析从 Config 到 Encoder三类任务共享同一个 XLM-RoBERTa 主干其 Rust 实现位于 candle-transformers/src/models/xlm_roberta.rs并已在 models/mod.rs 中公开导出。Config 结构xlm_roberta.rs对应 Hugging Faceconfig.json的关键字段hidden_size隐层维度num_attention_heads/intermediate_size注意力头数与 FFN 中间维度num_hidden_layers编码器层数vocab_size/max_position_embeddings/type_vocab_size词表、最大位置与 token type 数量layer_norm_eps/hidden_dropout_prob/attention_probs_dropout_prob归一化与 dropout 超参hidden_act中间层激活函数position_embedding_type位置编码类型pad_token_idpadding token id同时用于分词器 padding 与位置编码的 mask 计算Embeddings 层xlm_roberta.rs由word_embeddings、position_embeddings、token_type_embeddings与一个 LayerNorm 组成。值得注意的一个实现细节位置 id 不是直接使用绝对位置而是通过input_ids.ne(pad_token_id)构造 padding mask再做cumsum累加得到「相对位置 padding_idx 偏移」即非 padding token 依序编号确保 padding 区域不引入噪声位置编码。编码器xlm_roberta.rs按num_hidden_layers叠加XLMRobertaLayer每层包含多头自注意力query/key/value 三个线性层缩放因子1/sqrt(head_size)softmax_last_dim归一化、残差 LayerNorm 的 self-output、以及 dense 激活的 intermediate 层整体与经典 RoBERTa 一致。4D attention maskxlm_roberta.rs把[batch, seq_len]的二维 mask 先unsqueeze扩展到[batch, 1, tgt_len, src_len]再取反1.0 - expanded_mask并乘以 dtype 的最小值F32 时为f32::MIN这样被 mask 的位置在 softmax 前会被压到极小值注意力权重趋近于 0。运行与扩展建议首次运行会下载模型资源程序从 Hub 拉取tokenizer.json、config.json与model.safetensors或pytorch_model.bin。safetensors 以内存映射 F16 加载内存占用与加载速度都更友好。自定义本地资源若已提前下载好文件可通过--tokenizer-file、--config-file、--weight-files指向本地路径跳过自动下载也可用--model-id加载其他 XLM-RoBERTa 架构的模型。设备控制默认自动选 GPUCUDA/Metal 可用时纯 CPU 环境请加--cpu--release编译可显著提升推理性能。性能分析加--tracing会生成 trace 文件便于配合 tracing 工具定位瓶颈。改造接入三任务的输入文本都在 main.rs 中以常量数组硬编码实际业务中可将 reranker 的documents换成自己召回结果、将 text-classification 的sentences换成待分类文本即可模型加载、tokenize 与打分链路无需改动。本文介绍的三条命令覆盖了 XLM-RoBERTa 在检索重排、掩码语言建模与文本分类三类场景的完整链路结合示例入口与模型实现源码读者既可以开箱复现也能按需定制自己的推理应用。赞分享人工智能大模型机器学习深度学习本地部署模型推理服务【免费下载链接】candleMinimalist ML framework for Rust项目地址https://gitcode.com/GitHub_Trending/ca/candle点击查看免费下载相关推荐Plano 配置参考详解 plano_config.yml 完整字段与网关行为控制Plano 配置参考详解 plano_config.yml 完整字段与网关行为控制本指南是 Plano 网关单实例配置文件 plano_config.yml人工智能大模型机器学习深度学习本地部署模型推理服务在 Candle 中运行 DebertaV2/V3 微调模型NER 与文本分类实战指南在 Candle 中运行 DebertaV2/V3 微调模型NER 与文本分类实战指南导读 debertav2 示例 https://link.gitcod人工智能大模型机器学习深度学习本地部署模型推理服务如何 10 分钟搭好 Sunshine 游戏串流服务器Moonlight 新手完整指南如何 10 分钟搭好 Sunshine 游戏串流服务器Moonlight 新手完整指南想在客厅沙发上一拿手柄就能直接开玩 PC 里的游戏只需在主机上装好音视频后端上一篇ExifToolGui完全指南让照片元数据编辑变得简单高效下一篇Grizzly配置秘籍Contexts与环境变量设置轻松管理多环境监控资源创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

Candle 运行 XLM-RoBERTa 实战：Fill-Mask、Reranker 与文本分类三大任务指南

相关推荐

Univer SDK：嵌入式 Office 能力原子化实践指南

Ubuntu 22.04下Realtek R8125 2.5G网卡驱动安装与DKMS配置完整指南

Confluence表格操作全攻略：从基础建表到高级宏实战

更多资讯

Claude Code Skills实战：从SKILL.md编写到多技能组合落地

AI专著撰写新方法，借助AI工具1周完成20万字专著，高效又轻松！

基于2514张VOC数据集的摩托车电动车头盔检测YOLOv8实战

AI写专著实操指南掌握AI专著撰写方法高效产出20万字出版级专著

AI写专著实用攻略，借助AI专著生成工具快速搞定20万字合规专著

yolov5人群计数与阈值报警：从Coco预训练到工程落地的完整方案

今日更新

企业AI转型实战指南：从场景选择到落地避坑的完整路线图

OpenRig：本地大模型服务编排的轻量级运行时框架

夸克网盘1TB免费扩容领取全攻略：新老用户实操流程与避坑指南

本周精选

如何彻底离线使用 Laya-CoreML：Hugging Face 下载、缓存机制与 local_files_only 详解

基于1Panel 部署MySQL MCP 服务：TaoToken 统一 Key 接入与 config.toml 配置实战

崇州园区营销网站建设避坑指南：5大注意事项保备案安全

本月热门

我发现了一个新思路：用 Remotion + Claude Code 像写代码一样自动化生成短视频

Windows下 Codex 中 Chrome 和 Computer Use 插件不可用问题排查及解决参考方式：TaoToken 统一 Key 配置与验证

2026 大模型集体涨价：用 Python 做企业 Token 成本测算与选型避坑（附配置）

读完文章，想聊聊您的网站？