中文

通过多臂赌博机在近似线性时间内计算 medoid

机器学习 2017-11-08 v3 数据结构与算法 信息论 机器学习 math.IT

摘要

在高维空间中计算大量点的 medoid 是许多数据科学问题中日益常见的操作。我们提出一种算法 Med-dit,其使用 O(n log n) 次距离评估以高概率计算 medoid。Med-dit 基于与多臂赌博机问题的联系。我们在 Netflix-prize 和单细胞 RNA-Seq 数据集上实证评估了 Med-dit 的性能,这些数据集包含数十万个位于数万维空间中的点,并观察到相较当前最优方法 5-10 倍的性能提升。Med-dit 可在 https://github.com/bagavi/Meddit 获取。

关键词

引用

@article{arxiv.1711.00817,
  title  = {Medoids in almost linear time via multi-armed bandits},
  author = {Vivek Bagaria and Govinda M. Kamath and Vasilis Ntranos and Martin J. Zhang and David Tse},
  journal= {arXiv preprint arXiv:1711.00817},
  year   = {2017}
}