├── .DS_Store ├── LICENSE.txt ├── QExpand ├── __pycache__ │ └── prompts.cpython-310.pyc ├── cache │ ├── cache_qwen3_4b_med_and_rader │ │ └── doc_emb │ │ │ └── qwen3_4b_med_and_rader │ │ │ └── biology │ │ │ └── long_False │ │ │ └── 0.npy │ └── format_example.txt ├── output │ └── diver-qexpand │ │ └── biology │ │ ├── diver_aug0_result_retrieval.expand_query.json │ │ ├── diver_aug0_result_retrieval.metrics.json │ │ ├── diver_aug1_result_retrieval.expand_query.json │ │ ├── diver_aug1_result_retrieval.metrics.json │ │ ├── diver_aug2_result_retrieval.expand_query.json │ │ └── diver_aug2_result_retrieval.metrics.json ├── prompts.py ├── qexpand_main.py ├── run_qexpand.sh └── vllm_server │ ├── __pycache__ │ └── vllm_completion.cpython-310.pyc │ └── vllm_completion.py ├── README.md ├── README_CN.md ├── Reranker ├── groupwise_sft_data_example.jsonl ├── rerank_groupwise.py ├── rerank_listwise.py ├── rerank_merge_point_and_list.py ├── rerank_pointwise.py ├── reranker_script.sh └── train_sft_groupwise_reranker.sh ├── Retriever ├── merge_scores.py ├── prompts.py ├── retriever_script.sh ├── retrievers.py └── run.py ├── data └── BRIGHT │ └── readme.md ├── env_requirements ├── README.md ├── pytrec_eval-master.zip ├── requirements.txt └── trec_eval-9.0.8.tar.gz ├── pic ├── diver-bigpic-8-08251537.jpg ├── diver-bigpic-8-08251537.png ├── overview_DIVER.pdf ├── overview_DIVER.png └── rag_wechat.JPG ├── run_all.sh └── utils ├── __init__.py ├── common.py └── eval_util.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/.DS_Store -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /QExpand/__pycache__/prompts.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/__pycache__/prompts.cpython-310.pyc -------------------------------------------------------------------------------- /QExpand/cache/cache_qwen3_4b_med_and_rader/doc_emb/qwen3_4b_med_and_rader/biology/long_False/0.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/cache/cache_qwen3_4b_med_and_rader/doc_emb/qwen3_4b_med_and_rader/biology/long_False/0.npy -------------------------------------------------------------------------------- /QExpand/cache/format_example.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug0_result_retrieval.expand_query.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/output/diver-qexpand/biology/diver_aug0_result_retrieval.expand_query.json -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug0_result_retrieval.metrics.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/output/diver-qexpand/biology/diver_aug0_result_retrieval.metrics.json -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug1_result_retrieval.expand_query.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/output/diver-qexpand/biology/diver_aug1_result_retrieval.expand_query.json -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug1_result_retrieval.metrics.json: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug2_result_retrieval.expand_query.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/output/diver-qexpand/biology/diver_aug2_result_retrieval.expand_query.json -------------------------------------------------------------------------------- /QExpand/output/diver-qexpand/biology/diver_aug2_result_retrieval.metrics.json: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /QExpand/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/prompts.py -------------------------------------------------------------------------------- /QExpand/qexpand_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/qexpand_main.py -------------------------------------------------------------------------------- /QExpand/run_qexpand.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/run_qexpand.sh -------------------------------------------------------------------------------- /QExpand/vllm_server/__pycache__/vllm_completion.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/vllm_server/__pycache__/vllm_completion.cpython-310.pyc -------------------------------------------------------------------------------- /QExpand/vllm_server/vllm_completion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/QExpand/vllm_server/vllm_completion.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/README.md -------------------------------------------------------------------------------- /README_CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/README_CN.md -------------------------------------------------------------------------------- /Reranker/groupwise_sft_data_example.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/groupwise_sft_data_example.jsonl -------------------------------------------------------------------------------- /Reranker/rerank_groupwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/rerank_groupwise.py -------------------------------------------------------------------------------- /Reranker/rerank_listwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/rerank_listwise.py -------------------------------------------------------------------------------- /Reranker/rerank_merge_point_and_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/rerank_merge_point_and_list.py -------------------------------------------------------------------------------- /Reranker/rerank_pointwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/rerank_pointwise.py -------------------------------------------------------------------------------- /Reranker/reranker_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/reranker_script.sh -------------------------------------------------------------------------------- /Reranker/train_sft_groupwise_reranker.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Reranker/train_sft_groupwise_reranker.sh -------------------------------------------------------------------------------- /Retriever/merge_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Retriever/merge_scores.py -------------------------------------------------------------------------------- /Retriever/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Retriever/prompts.py -------------------------------------------------------------------------------- /Retriever/retriever_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Retriever/retriever_script.sh -------------------------------------------------------------------------------- /Retriever/retrievers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Retriever/retrievers.py -------------------------------------------------------------------------------- /Retriever/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/Retriever/run.py -------------------------------------------------------------------------------- /data/BRIGHT/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/data/BRIGHT/readme.md -------------------------------------------------------------------------------- /env_requirements/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/env_requirements/README.md -------------------------------------------------------------------------------- /env_requirements/pytrec_eval-master.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/env_requirements/pytrec_eval-master.zip -------------------------------------------------------------------------------- /env_requirements/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/env_requirements/requirements.txt -------------------------------------------------------------------------------- /env_requirements/trec_eval-9.0.8.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/env_requirements/trec_eval-9.0.8.tar.gz -------------------------------------------------------------------------------- /pic/diver-bigpic-8-08251537.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/pic/diver-bigpic-8-08251537.jpg -------------------------------------------------------------------------------- /pic/diver-bigpic-8-08251537.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/pic/diver-bigpic-8-08251537.png -------------------------------------------------------------------------------- /pic/overview_DIVER.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/pic/overview_DIVER.pdf -------------------------------------------------------------------------------- /pic/overview_DIVER.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/pic/overview_DIVER.png -------------------------------------------------------------------------------- /pic/rag_wechat.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/pic/rag_wechat.JPG -------------------------------------------------------------------------------- /run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/run_all.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/eval_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AQ-MedAI/Diver/HEAD/utils/eval_util.py --------------------------------------------------------------------------------