Skip to main navigation Skip to search Skip to main content

Federated Retrieval Over Embedding-Heterogeneous Vector Databases

  • Yuxiang Wang
  • , Yongxin Tong
  • , Zimu Zhou
  • , Ziyuan He
  • , Ruixi Hu
  • , Ke Xu

Research output: Chapters, Conference Papers, Creative and Literary Works › RGC 32 - Refereed conference paper (with host publication) › peer-review

Abstract

Vector databases are increasingly used to manage unstructured data by mapping them into high-dimensional embeddings and enabling efficient similarity retrieval. Many realworld applications, such as legal document retrieval and medical question answering, require embedding-based retrieval in federated environments where data are distributed across autonomous silos. We formulate this setting as Federated Approximate Nearest Neighbor Search (FANNS), where a server issues a query along with a query embedding model, aiming to retrieve the top- k nearest objects from datasets across all silos. A key challenge in FANNS is embedding heterogeneity, where silos and the server employ different embedding models, a problem overlooked in prior research. To address this challenge, we exploit the nonIID nature of federated data and propose two novel adaptive algorithms for FANNS queries. The first is a competition-based method that dynamically adjusts retrieval sizes across silos, though it can be sensitive to misleading candidates. The second is a contribution-based method that samples promising silos based on their accumulated contributions, and we provide theoretical guarantees on its latency reduction. Evaluations on four datasets show that our method achieves over 90% retrieval accuracy and 2. 3 × to 6. 2 × speedups over existing solutions. © 2026 IEEE.
Original languageEnglish
Title of host publicationProceedings - 2026 IEEE 42nd International Conference on Data Engineering (ICDE 2026)
PublisherIEEE
Pages350-363
Number of pages14
ISBN (Electronic)979-8-3315-8365-1
DOIs
Publication statusPublished - 2026
Event42nd International Conference on Data Engineering (ICDE 2026) - Montreal, Canada
Duration: 4 May 2026 → 8 May 2026

Publication series

NameProceedings - IEEE International Conference on Data Engineering, ICDE

Conference

Conference42nd International Conference on Data Engineering (ICDE 2026)
PlaceCanada
CityMontreal
Period4/05/26 → 8/05/26

Bibliographical note

Full text of this publication does not contain sufficient affiliation information. With consent from the author(s) concerned, the Research Unit(s) information for this record is based on the existing academic department affiliation of the author(s).

Funding

This work was partially supported by National Science Foundation of China (NSFC) (Grant Nos. 62425202, 62336003), National Key Research and Development Program of China under Grant No. 2023YFF0725103, the Beijing Natural Science Foundation (Z230001), the Fundamental Research Funds for the Central Universities No. JK2024-03, the Didi Collaborative Research Program and the State Key Laboratory of Complex & Critical Software Environment (SKLCCSE). Zimu Zhou s research is supported by National Science Foundation of China (NSFC) (No. 62572412), the RGC of Hong Kong SAR, China (Project No. CityU 11206425).

Research Keywords

  • approximate nearest neighbor search
  • high-dimensional data

Fingerprint

Dive into the research topics of 'Federated Retrieval Over Embedding-Heterogeneous Vector Databases'. Together they form a unique fingerprint.

Cite this