news 2026/4/16 10:56:09

3.6 Elasticsearch-深度学习排序:Learning to Rank 插件安装与特征工程

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
3.6 Elasticsearch-深度学习排序:Learning to Rank 插件安装与特征工程

3.6 Elasticsearch-深度学习排序:Learning to Rank 插件安装与特征工程

3.6.1 为什么要在 Elasticsearch 里做 Learning to Rank

传统 TF-IDF、BM25 这类词袋评分函数在长尾查询、语义漂移、多字段混合场景下很快遇到天花板。把深度学习模型直接丢进离线打分再灌回 ES 虽然简单,但延迟高、实时性差。Elasticsearch Learning to Rank(LTR)插件把「特征抽取 → 模型推理 → 结果重排序」整条链路搬到 ES 内部,毫秒级响应,同时保留倒排索引的召回优势,是工业界「粗排+精排」架构的标配。

3.6.2 插件版本对齐矩阵
ElasticsearchLTR 插件JDKPython 客户端
7.17.x7.17.0.011elasticsearch-ltr 1.3
8.11.x8.11.0.017elasticsearch-ltr 2.1

8.x 开始官方把transport-client完全移除,必须用 REST 低级客户端上传模型。

3.6.3 在线安装与集群滚动重启
  1. 每台 node 执行
    sudobin/elasticsearch-plugininstall\https://github.com/o19s/elasticsearch-learning-to-rank/releases/download/v8.11.0.0/ltr-8.11.0.0.zip
  2. 校验
    curl-XGET"localhost:9200/_ltr"|jq'.version'
  3. 滚动重启:每台 node 先/_cluster/nodes/_local/_shutdown等分片重分配完成再启下一台,保证绿色状态。
3.6.4 特征工程:从倒排到深度语义

LTR 把特征分成三类:

  • Query Feature:仅与查询相关,如查询长度、是否含品牌词。
  • Document Feature:仅与文档相关,如商品销量、发布时间。
  • Query-Document Feature:交叉特征,占模型 80% 以上权重。

下面给出电商搜索场景 18 维特征模板,可直接拷贝到ltr_feature_set.json

{"name":"ecommerce_features","params":["keywords"],"feature":[{"name":"title_bm25","class":"org.apache.lucene.search.Explanation","query":{"match":{"title":"{{keywords}}"}}},{"name":"category_match","class":"org.apache.lucene.search.Explanation","query":{"term":{"category":"{{keywords}}"}}},{"name":"brand_exact","class":"org.apache.lucene.search.Explanation","query":{"term":{"brand.keyword":"{{keywords}}"}}},{"name":"sales","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"field_value_factor":{"field":"sales","modifier":"log1p"}}}},{"name":"price","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"field_value_factor":{"field":"price"}}}},{"name":"in_stock","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"filter":{"term":{"stock":true}},"weight":1}}},{"name":"discount","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"Math.max(0.0, doc['marketPrice'].value - doc['price'].value) / doc['marketPrice'].value"}}}}},{"name":"title_length","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"doc['title'].value.length()"}}}}},{"name":"query_length","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"params._source.query.length()"}}}}},{"name":"title_embedding_dot","class":"org.apache.lucene.search.Explanation","query":{"script_score":{"script":{"source":"cosineSimilarity(params.queryVector, 'title_vector') + 1.0","params":{"queryVector":"{{query_vector}}"}}}}},{"name":"desc_embedding_dot","class":"org.apache.lucene.search.Explanation","query":{"script_score":{"script":{"source":"cosineSimilarity(params.queryVector, 'desc_vector') + 1.0","params":{"queryVector":"{{query_vector}}"}}}}},{"name":"recall_score","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"_score"}}}}},{"name":"click_ctr","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"field_value_factor":{"field":"click_ctr","modifier":"log1p"}}}},{"name":"cart_ctr","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"field_value_factor":{"field":"cart_ctr","modifier":"log1p"}}}},{"name":"pay_ctr","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"field_value_factor":{"field":"pay_ctr","modifier":"log1p"}}}},{"name":"freshness","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"Math.max(0, (params.now - doc['createTime'].value.getMillis()) / 86400000)","params":{"now":"{{now}}"}}}}}},{"name":"query_doc_jaccard","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"script_score":{"script":{"source":"double q = params.queryTerms.size(); double d = doc['title_terms'].size(); double i = 0; for (term in params.queryTerms) { if (doc['title_terms'].contains(term)) i++; } return i / (q + d - i + 1e-6);","params":{"queryTerms":"{{query_terms}}"}}}}}},{"name":"is_promotion","class":"org.apache.lucene.search.Explanation","query":{"function_score":{"filter":{"range":{"promotionStart":{"lte":"now"},"promotionEnd":{"gte":"now"}}},"weight":1}}}]}

上传命令:

curl-XPUT"localhost:9200/_ltr/_featureset/ecommerce_features"\-H'Content-Type: application/json'--data @ltr_feature_set.json
3.6.5 深度语义特征实时注入

title_embedding_dot依赖向量字段,需要在 mapping 里显式声明:

"title_vector":{"type":"dense_vector","dims":384,"similarity":"cosine"}

查询时把离线微调的 MiniLM 向量作为query_vector参数传进来即可,无需二次分词,延迟 <5 ms。

3.6.6 特征日志采样与存储

训练数据通过sltr查询生成,样例:

GET/products/_search{"query":{"match":{"title":"iphone 15"}},"rescore":{"window_size":100,"query":{"rescore_query":{"sltr":{"params":{"keywords":"iphone 15","query_vector":[...],"now":1700000000000},"featureset":"ecommerce_features","store":true,"logging":true}}}}}

ES 会把 18 维特征值写入.ltrstore索引,字段_ltrlog可直接拉下来做 LibSVM 格式转换:

curl-XGET"localhost:9200/.ltrstore/_search?q=_ltrlog:*"\|jq -r'.hits.hits[]._source._ltrlog'>train.svmlight
3.6.7 模型训练与上传

XGBoost 示例:

importxgboostasxgb dtrain=xgb.DMatrix('train.svmlight')params={'objective':'rank:pairwise','eta':0.1,'max_depth':6}bst=xgb.train(params,dtrain,num_boost_round=300)bst.save_model('xgb_model.json')

上传:

curl-XPUT"localhost:9200/_ltr/_model/xgb_model"\-H'Content-Type: application/json'\--data-binary @xgb_model.json
3.6.8 线上 A/B:粗排 + LTR 精排
"rescore":{"window_size":200,"query":{"rescore_query":{"sltr":{"model":"xgb_model","params":{"keywords":"{{keywords}}","query_vector":"{{query_vector}}","now":"{{now}}"}}},"query_weight":0,"rescore_query_weight":1}}

window_size决定粗排截断位置,线上实验表明 200 条召回再精排,点击收益 +8.7%,P99 延迟仅增加 12 ms。

3.6.9 性能调优清单
  1. 特征缓存:把sales、price等静态特征拆到function_scoreweight里,ES 会缓存 DocValues,避免重复计算。
  2. 向量量化:384 维 float32 → 8 位整型,内存降 4 倍,精度下降 <0.5%。
  3. 线程池隔离:给searchml.utility单独线程池,防止大促期间互相挤占。
  4. 模型热更新:利用_ltr/_model/{name}/_update接口,灰度 5% 节点先加载,QPS 无抖动。
3.6.10 常见坑
  • 8.x 以后rank_evalAPI 默认跳过 rescore,需要显式加?search_type=dfs_query_then_fetch
  • dense_vector字段不支持doc_values,做特征日志时务必用store: true把向量存_source,否则拉不到值。
  • 若用rank:ndcg训练,上传模型前把 XGBoost 的base_score置 0,不然 ES 会多累加一次偏置,导致打分整体漂移。

至此,Elasticsearch 侧的深度学习排序链路全部打通:插件安装 → 特征工程 → 模型训练 → 线上热加载 → A/B 实验。下一节将介绍如何把用户实时行为流(点击、加购、支付)通过 Flink CEP 拼接成样本,实现「模型日更」的闭环。
更多技术文章见公众号: 大城市小农民

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/15 14:13:06

基于Java+ vue校务管理系统(源码+数据库+文档)

校务管理 目录 基于springboot vue校务管理系统 一、前言 二、系统功能演示 三、技术选型 四、其他项目参考 五、代码参考 六、测试参考 七、最新计算机毕设选题推荐 八、源码获取&#xff1a; 基于springboot vue校务管理系统 一、前言 博主介绍&#xff1a;✌️大…

作者头像 李华
网站建设 2026/4/16 10:53:59

Stop-motion-OBJ插件完全指南:从零开始制作精美3D动画

还在为Blender中导入网格序列而烦恼吗&#xff1f;&#x1f914; Stop-motion-OBJ插件让这一切变得简单&#xff01;这个强大的工具能帮你轻松导入OBJ、STL、PLY等格式的网格文件&#xff0c;将它们转化为流畅的3D动画。无论你是新手还是专业人士&#xff0c;这份指南都将带你快…

作者头像 李华
网站建设 2026/4/14 21:52:34

R.swift升级实战:从旧版本到新架构的完整迁移手册

R.swift升级实战&#xff1a;从旧版本到新架构的完整迁移手册 【免费下载链接】R.swift Strong typed, autocompleted resources like images, fonts and segues in Swift projects 项目地址: https://gitcode.com/gh_mirrors/rs/R.swift 在Swift项目开发中&#xff0c;…

作者头像 李华
网站建设 2026/4/10 14:17:47

WindowResizer终极指南:5分钟掌握窗口尺寸强制调整技巧

WindowResizer终极指南&#xff1a;5分钟掌握窗口尺寸强制调整技巧 【免费下载链接】WindowResizer 一个可以强制调整应用程序窗口大小的工具 项目地址: https://gitcode.com/gh_mirrors/wi/WindowResizer 你是否曾经遇到过那些顽固的应用程序窗口&#xff1f;它们要么尺…

作者头像 李华
网站建设 2026/4/14 15:05:42

66、XML解析:SAX与DOM方法全解析

XML解析:SAX与DOM方法全解析 1. 使用SAX解析XML 在大多数情况下,从XML文档中提取信息的最佳方式是使用符合SAX(Simple API for XML)标准的事件驱动解析器来解析文档。SAX定义了一个标准API,可以在许多不同的底层解析器之上实现。SAX解析方法与大多数HTML解析器有相似之处…

作者头像 李华
网站建设 2026/4/16 0:10:35

Peek:终极简单的Linux动态GIF录制神器

Peek&#xff1a;终极简单的Linux动态GIF录制神器 【免费下载链接】peek Simple animated GIF screen recorder with an easy to use interface 项目地址: https://gitcode.com/gh_mirrors/pe/peek 还在为制作屏幕演示而烦恼吗&#xff1f;&#x1f914; Peek是一款专为…

作者头像 李华