本仓库实现了一篇未开源论文《Swarm Intelligence Enhanced Reasoning: A Density-Driven Framework for LLM-Based Multi-Agent Optimization》的代码。原文链接
想要运行程序,推荐的python版本是3.10,并需要做如下准备工作。 在终端运行:
- 安装依赖库:
pip install -r requirements.txt
- 配置PRM模型文件:
huggingface-cli download Qwen/Qwen2.5-Math-PRM-7B --local-dir ./PRM --resume-download
- 获取基准测试数据集(AIME2025建议去官网手动下载):
python load_dataset.py
工程文件的用途解释:
gpt_structure.py调用openai的大模型接口,需要你提前在系统的环境变量中配置好key和base_url。nds.py非支配排序算法。kde_utils.py核密度计算。evaluator.py评估器,基于qwen2.5的PRM模型。generator.py生成器,使用ZeroShot的CoT进行逐步推理。sier.py密度辅助算法的主程序。cluster.py用于标签和种群聚类。