Skip to content

ashcrow123/SIA

Repository files navigation

本仓库实现了一篇未开源论文《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用于标签和种群聚类。

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages