Neural Conversational Search Baselines for TREC iKAT

This repository provide baselines and tools for the TREC iKAT track: the Interactive Knowledge Assistance Track at TREC. This track focuses on neural conversational search with user personalization, context, and goal-driven information seeking.

The TREC iKAT track models goal-oriented conversations where a user interacts with a search agent to complete a complex task (e.g., finding a university, preparing for a trip, obtaining a hunting license). The agent can use both dialogue history and user profile information (PTKB) to personalize and adapt its retrieval. See guidelines for more details https://www.trecikat.com/guidelines/

Example of information seeking conversation:

PTKB: I live in the Netherlands. I have a bachelor’s degree in computer science from Tilburg University.

[User]: I want to start my master’s degree, can you help me with finding a university?
[System]: Do you want to study abroad?
[User]: No, I don't want to go abroad.
[System]: I can help you find a university to continue your studies in the Netherlands as a computer science student. Take a look at these Top Computer Science Universities in the Netherlands: 1. Delft University of Technology. 2. Eindhoven University of Technology 3. Vrije Universiteit Amsterdam.
[User]: ...

We provide in this repository:

• SPLADE-based retrieval [SPLADE Retrieval]
• Query rewriting with PTKB integration (manual and GPT-based)
• BM25 baselines [BM25 Retrieval]
• Cross-encoder reranking (MiniLM and DeBERTa) [Reranking]
• Evaluation of the runs [Performance]

Installation

Conda Environment

conda env create -f environment.yml
conda activate ikat24

Data

Download the iKAT dataset:

bash ./DATA/dl_ikat.sh

Provide example of ikat turn with ptkb

Also required:

• SPLADE NumBa index (see DATA/README.md)
• Pyserini Lucene index (see DATA/README.md)

SPLADE retrieval

Requires 256GB RAM and one GPU.

Running SPLADE retrieval on human manual rewritten queries (both iKAT 2023 and iKAT 2024):

export SPLADE_CONFIG_NAME="config_hf_splade_ikat.yaml"

index_dir=/ivi/ilps/projects/ikat24/splade_index_website/splade_index/
eval_queries=[DATA/queries_manual_2023.tsv,DATA/queries_manual_2024.tsv]
out_dir=EXP/manual_splade

python -m splade.retrieve init_dict.model_type_or_dir=naver/splade-cocondenser-ensembledistil \
        config.pretrained_no_yamlconfig=true config.index_dir=$index_dir \
        config.out_dir=$out_dir \
        data.Q_COLLECTION_PATH=$eval_queries 

Now using a GPT4o-mini rewrite to integrate the PTKB and generate the rewrite:

export OPENAI_API_KEY="your_openai_key"

python DATA/rewrite_gpt_ikat.py

Now running SPLADE on the rewrite:

export SPLADE_CONFIG_NAME="config_hf_splade_ikat.yaml"

index_dir=/ivi/ilps/projects/ikat24/splade_index_website/splade_index/
eval_queries=[DATA/queries_gpt4o_2023.tsv,DATA/queries_gpt4o_2024.tsv]
out_dir=EXP/gpt4o_splade

python -m splade.retrieve init_dict.model_type_or_dir=naver/splade-cocondenser-ensembledistil \
        config.pretrained_no_yamlconfig=true config.index_dir=$index_dir \
        config.out_dir=$out_dir \
        data.Q_COLLECTION_PATH=$eval_queries

BM25 retrieval

Similarly you can do retrieval with BM25, this requires less ressource.

With gpt4o-mini rewrite (on iKAT 2024):

python -m bm25.retrieve --index_path /ivi/ilps/projects/TREC-Ikat-CW22/passage_index/trec_ikat_2023_passage_index \
                        --topics DATA/queries_gpt4o_2024.tsv \
                        --output EXP/gpt4o_bm25/IKAT2024/run.json

You can evaluate the runs with:

For evaluate we use binary qrels for recall and mrr, and the graded relevance for ndcg.

python -m splade.evaluate --run_dir EXP/gpt4o_bm25/IKAT2024/run.json \
                          --qrel_file_path DATA/2024_test_qrels.json \
                          --qrel_binary_file_path DATA/2024_test_qrels_binary.json

Reranking

For reranking we also use the lucene index, to load the text from the retrieved documents id.

python -m rerank.rerank --index_path /ivi/ilps/projects/TREC-Ikat-CW22/passage_index/trec_ikat_2023_passage_index \
                        --model naver/trecdl22-crossencoder-debertav3 \
                        --run EXP/gpt4o_splade/IKAT2024/run.json \
                        --query_file DATA/queries_gpt4o_2024.tsv \
                        --output EXP/gpt4o_splade_rerank/IKAT2024/run.json

python -m rerank.rerank --index_path /ivi/ilps/projects/TREC-Ikat-CW22/passage_index/trec_ikat_2023_passage_index \
                        --model naver/trecdl22-crossencoder-debertav3 \
                        --run EXP/gpt4o_bm25/IKAT2024/run.json \
                        --query_file DATA/queries_gpt4o_2024.tsv \
                        --output EXP/gpt4o_bm25_rerank/IKAT2024/run.json

can use cross-encoder/ms-marco-MiniLM-L-6-v2, or run it on iKAT 2023.

Similarly you can evaluate the produced runs:

python -m splade.evaluate --run_dir EXP/gpt4o_splade_rerank/IKAT2024/run.json \
                          --qrel_file_path DATA/2024_test_qrels.json \
                          --qrel_binary_file_path DATA/2024_test_qrels_binary.json

Performance

We provide here a brief summary of several produced runs:

IKAT23 Results

Note that the pool of assessed passages is biased toward BM25.

IKAT24 Results

SPLADE interactive retrieval

A iKAT searcher interactive tool. TBD.

---

Resources

• Repository for ANCE dense retrieval on iKAT [code]

• TREC iKAT webiste: https://www.trecikat.com/

• Official TREC website: https://trec.nist.gov/

• Contact: s.c.lupart@uva.nl