π€ AI Summary
Existing LLM-based embedders apply uniform query augmentation to all inputs, increasing latency and degrading performance on certain queries; no existing method supports adaptive query augmentation in multimodal settings. Method: We propose M-Solomonβthe first general-purpose multimodal embedding model enabling adaptive query augmentation. It leverages conditional prefixes (/augment or /embed) to guide a multimodal large language model (MLLM) in dynamically determining whether a given query requires augmentation, generating synthetic augmented content only when necessary. Fine-grained control is achieved via prefix-guided generation and a two-stage training strategy. Results: Experiments demonstrate that M-Solomon significantly outperforms both non-augmented and uniformly augmented baselines, improving embedding quality while maintaining low latency. To our knowledge, it is the first approach to achieve efficient and precise adaptive augmentation in multimodal embedding.
π Abstract
Query augmentation makes queries more meaningful by appending further information to the queries to find relevant documents. Current studies have proposed Large Language Model (LLM)-based embedders, which learn representation for embedding and generation for query augmentation in a multi-task manner by leveraging the generative capabilities of LLM. During inference, these jointly trained embedders have conducted query augmentation followed by embedding, showing effective results. However, augmenting every query leads to substantial embedding latency and query augmentation can be detrimental to performance for some queries. Also, previous methods have not been explored in multimodal environments. To tackle these problems, we propose M-Solomon, a universal multimodal embedder that can adaptively determine when to augment queries. Our approach first divides the queries of the training datasets into two groups at the dataset level. One includes queries that require augmentation and the other includes queries that do not. Then, we introduces a synthesis process that generates appropriate augmentations for queries that require them by leveraging a powerful Multimodal LLM (MLLM). Next, we present adaptive query augmentation. Through this step, M-Solomon can conduct query augmentation only when necessary by learning to generate synthetic augmentations with the prefix /augment for queries that demand them and to generate the simple string /embed for others. Experimental results showed that M-Solomon not only surpassed the baseline without augmentation by a large margin but also outperformed the baseline that always used augmentation, providing much faster embedding latency.