Sliced Wasserstein + 手写 HNSW:4.8 万幅名画的配色搜索引擎

Potential-Barber8658 · reddit · 2026-10-11

Palette Atlas 项目实现了按配色方案检索 48,695 幅公有领域绘画。核心是把每幅图压缩为 128 个等权颜色,自然距离是 Earth Mover's Distance(EMD),但逐对计算约 6ms,无法做近邻搜索。

嵌入方法:把 OKLab 颜色投影到 Fibonacci 半球的 8 个方向上,排序后取 16 分位平均,两向量间的 L1 即各方向 1D W1 的均值,是真实 W1 的下界。

效果对比(240 查询,与精确 EMD 的 top10 重合率/距离偏差)

索引:从零手写 HNSW(M=16, efConstruction=200),ef=32 时 recall@10 达 1.000,单查询 0.25ms,暴力搜索需 14ms。向量内在维度约 11,层级结构仍然有效。对论文的一处改动:顶层用 8 个 k-means 地标节点替代随机节点,搜索成本不变(平均 2,558 vs 2,556 次距离计算),顶层变成每个色系一幅代表画。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →