Efficient legal case search method speeds up retrieving relevant cases
Cassette: Case-to-Case Structural Distillation for Efficient Legal Case Retrieval
Information Retrieval
Summary
Finding important legal cases fast is hard because comparing all of them takes a very long time. The authors created a new method called Cassette that learns from a powerful but slow system to train a faster, simpler model. This new model uses smart encoding techniques that work quickly online and prepare candidate cases offline. Tests show this approach keeps accuracy high while making case search much faster. The code for this method is publicly available for others to use.
legal case retrievalgraph neural networksknowledge distillationranking objectivemultilayer perceptrondual encoderquery encodercandidate encodercomputational efficiency
Authors
Yanran Tang, Ruihong Qiu, Hongzhi Yin, Xue Li, Zi Huang
Abstract
Legal case retrieval (LCR) is an essential tool for not only assisting legal practitioners to efficiently retrieve precedents but also enabling ordinary individuals to find valuable legal case information without relying on expensive professional legal services. Our previous work CaseLink demonstrated the effectiveness of using case to case graph structures to improve retrieval accuracy. However, its high computational cost during inference on large-scale legal databases limits its practical use in real-world settings. The main inefficiency comes from constructing test time graphs and computing pairwise term frequency similarities of cases. This process has O(n^2) complexity for n legal cases, making the runtime prohibitive as the number of candidates grows. For example, the retrieval time for one query on a database (COLIEE2022) with 1,563 candidate cases is more than 500 milliseconds, while the runtime would increase drastically to more than 3,500 seconds for a database (LeCaRDv2) with 55,192 candidate cases. To further enhance the retrieval performance while achieving a significant speed-up, in this extension paper, Cassette framework is proposed with a distillation strategy involving ranking objective and eigen-matching objective for an effective transfer of knowledge from a powerful and well-trained heavy teacher retriever to a lightweight and efficient hybrid student dual encoder. Specifically, the student query encoder is implemented as a multilayer perceptron model designed for fast online processing, whereas the student candidate encoder adopts a GNN architecture, suitable for an offline manner within the case database. Extensive experiments are conducted on three benchmark datasets, and the results verify the effectiveness of the ranking distillation while achieving high efficiency. The code has been released on https://github.com/yanran-tang/Cassette/.