Subspace Representations for Soft Set Operations and Sentence Similarities
Yoichi Ishibashi, Sho Yokoi, Katsuhito Sudoh, Satoshi Nakamura
TL;DR
This work proposes representing word sets as linear subspaces within pretrained embedding spaces, grounded in quantum logic, to enable robust set operations (union, intersection, complement) and soft membership. By replacing hard, binary membership with a subspace indicator function and extending sentence similarity metrics (BERTScore) to SubspaceBERTScore, the approach captures richer semantic structure and improves performance on semantic textual similarity and set retrieval tasks. The method requires no additional training and leverages existing embeddings, offering a principled way to manipulate word sets and measure their similarity. Empirical results across STS benchmarks and a set-retrieval dataset show consistent gains over traditional vector-based and fuzzy-set methods, highlighting practical implications for NLP tasks involving concept sets and sentence understanding.
Abstract
In the field of natural language processing (NLP), continuous vector representations are crucial for capturing the semantic meanings of individual words. Yet, when it comes to the representations of sets of words, the conventional vector-based approaches often struggle with expressiveness and lack the essential set operations such as union, intersection, and complement. Inspired by quantum logic, we realize the representation of word sets and corresponding set operations within pre-trained word embedding spaces. By grounding our approach in the linear subspaces, we enable efficient computation of various set operations and facilitate the soft computation of membership functions within continuous spaces. Moreover, we allow for the computation of the F-score directly within word vectors, thereby establishing a direct link to the assessment of sentence similarity. In experiments with widely-used pre-trained embeddings and benchmarks, we show that our subspace-based set operations consistently outperform vector-based ones in both sentence similarity and set retrieval tasks.
