lmcinnes / lmcinnes/umap

Combining multiple UMAP for different embedding models

Open
#585 4 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
8.3k
Forks
871
Avg merge
1d 13h
Merged PRs (30d)
5

Description

https://umap-learn.readthedocs.io/en/latest/composing_models.html
This tutorial shows how 2 views of same data can be combined. I was wondering if this would work on 2 different embeddings on the same data?
Usecase -- We often cluster using tsne or umap using representation embeddings from some ML model. And often times we have 2 different embeddings capturing different features of the same data. One way to combine them for similarity is
w1*sim1 + w2*sim2 and use that as a combined similarity score that capture both emb1 and emb2 feature similarity together.
Made me wonder if combining through umap would have an advantage to cluster data which will capture features from both embedding models.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the composing models tutorial linked in the issue and assess whether its approach applies to two different embeddings of the same data. Compare that with the proposed weighted similarity combination, and define whether a reproducible example, documentation, or implementation is needed to establish the benefit for clustering.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.