1. python==3.8

2. pip install -r src/requirements.txt

3. Download data

   This repo contains a modified version of CoSearcher.

   cd cosearcher
   scripts/bootstrap.sh
   

   (Some minor errors may occur at this step.)

1. Preprocess data and generate train/val/test split with long/short cooperativeness split.

   cd src

   Run data_process.ipynb cell 1

2. Train the RoBERTa classifier for answer type prediction.

   Run data_process.ipynb cell 2 and cell 3

3. Predict answer type with trained RoBERTa classifier.

   Run data_process.ipynb cell 4

4. Finetune T5 and UnifiedQA (on Qulac for example)

   cd src
   deepspeed t5trainer.py \
     --deepspeed ds_config_zero3.json \
     --model_name_or_path t5-small \
     --do_train \
     --do_eval \
     --do_predict \
     --source_prefix "answer: " \
     --output_dir output/t5-small-qulac/ \
     --per_device_train_batch_size=64 \
     --per_device_eval_batch_size=64 \
     --overwrite_output_dir \
     --predict_with_generate \
     --text_column t5-question \
     --summary_column answer \
     --seed 2023 \
     --num_train_epochs 30 \
     --train_file ../data/processed/qulac_train.csv \
     --validation_file ../data/processed/qulac_dev.csv \
     --test_file ../data/processed/qulac_test.csv 
   

   deepspeed t5trainer.py \
     --deepspeed ds_config_zero3.json \
     --model_name_or_path allenai/unifiedqa-t5-small \
     --do_train \
     --do_eval \
     --do_predict \
     --source_prefix "" \
     --output_dir output/unifiedqa-small-qulac/ \
     --per_device_train_batch_size=64 \
     --per_device_eval_batch_size=64 \
     --overwrite_output_dir \
     --predict_with_generate \
     --text_column unifiedqa-question \
     --summary_column answer \
     --seed 2023 \
     --num_train_epochs 30 \
     --train_file ../data/processed/qulac_train.csv \
     --validation_file ../data/processed/qulac_dev.csv \
     --test_file ../data/processed/qulac_test.csv 
   
   

5. Process the generated responses from LMs and save it to csv for evaluation.

   Run data_process.ipynb cell 5 with the test_file and output_dir from the last step.

6. Run document retrieval experiments using the output_csv from the last step. (with post-processed t5 results for example)

   cd cosearcher
   python3 src/main.py --dataset qulac --output_file_path ../src/output/t5-small-qulac.csv > ../src/output/t5-small-qulac.json
   

   Make sure to change dataset to clariq for document retrieval experiments on ClariQ.

7. Run evaluations in view.ipynb

8. Significance test in sigtest.py

9. Human evaluation instructions and results are available in data/crowd

Please cite the following work if you use this code repository in your work:

@misc{wang2024indepth,
      title={An In-depth Investigation of User Response Simulation for Conversational Search}, 
      author={Zhenduo Wang and Zhichao Xu and Qingyao Ai and Vivek Srikumar},
      year={2024},
      eprint={2304.07944},
      archivePrefix={arXiv},
      primaryClass={cs.IR}
}