TensorSearch: Parallel Similarity Search on Tensors
Existing similarity search methods, often limited to scalar or vector data, struggle to identify complex patterns found in scientific datasets, such as 2D seismic events or 3D magnetic flux ropes. We introduce TensorSearch, a novel parallel similarity search paradigm designed to identify known patterns in high-dimensional tensors. By directly employing tensor representations, TensorSearch captures intricate pattern structures more effectively than traditional vector-based approaches. Furthermore, its parallel architecture optimizes cache and I/O operations, enabling efficient processing of large-scale scientific data. Our performance evaluations demonstrate that TensorSearch outperforms state-of-the-art vector-based systems like Milvus by up to 10x, and achieves up to a remarkable 55x advantage over custom solution developed in Matlab used by the domain scientists. In these tests, TensorSearch exhibits linear scalability, supporting up to 2240 CPU cores.
Authors
- Avinash Nayak (ORCID: https://orcid.org/0000-0001-7913-7189)
- Verónica Rodrı́guez Tribaldos (ORCID: https://orcid.org/0000-0002-1325-0434)
- Kesheng Wu (ORCID: https://orcid.org/0000-0002-6907-3393)
- Jonathan Ajo‐Franklin (ORCID: https://orcid.org/0000-0002-6666-4702)
- Suren Byna (ORCID: https://orcid.org/0000-0003-3048-3448)
- Alex Sim (ORCID: https://orcid.org/0000-0002-6295-1982)
- Patrick Dobson (ORCID: https://orcid.org/0000-0001-5031-8592)
- Qile Zhang (ORCID: https://orcid.org/0000-0002-9504-641X)
- Fan Guo (ORCID: https://orcid.org/0000-0003-4315-3755)
- Bin Dong (ORCID: https://orcid.org/0000-0002-0725-0833)
Institutions
- Los Alamos National Laboratory (US)
- Lawrence Berkeley National Laboratory (US)
- GFZ Helmholtz Centre for Geosciences (DE)
- The Ohio State University (US)
- Rice University (US)
Publication Details
- Published
- 2024-12-15
- DOI
- https://doi.org/10.1109/bigdata62323.2024.10825798
- Primary Topic
- Computational Physics and Python Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- U.S. Department of Energy
- Office of Energy
- Office of Science
- Technology Development
- Geothermal Technologies Office