arXiv:2607.01366cs.AI2026-07

用智能代理自动搜索联邦学习算法组合,提升医疗数据协作建模效果

Auto-FL-Research: Agentic Search for Federated Learning Algorithms

论文配图:Auto-FL-Research: Agentic Search for Federated Learning Algorithms
图 1 · 摘自论文原文
  • 构建可自主设计联邦学习算法的智能体工作流,支持多种训练策略组合
  • 在五个医疗联邦数据集上实现平均性能提升,部分结果稳定可复现
  • 适合研究联邦学习算法设计、需高效探索超参数与架构的开发者

联邦学习研究常依赖大量细微但关键的算法选择:优化器变体、服务器聚合规则、本地训练调度、归一化与正则化方式及模型结构。这些选择手动探索成本高,且不同修改可能改变训练或评估路径,难以公平比较。本文提出Auto-FL-Research(AFR),一种受限编码智能体工作流,用于搜索联邦学习算法配方。智能体可提出并实现包含服务器聚合规则、客户端更新调度、本地目标函数和注册模型变体在内的候选算法,任务配置固定变异空间、计算预算、通信协议和最终模型评估标准。每轮实验记录得分、运行时间、修改文件、生成产物及失败状态。我们在五个医疗跨孤岛联邦学习数据集(FLamby)和五个固定客户端分组的LEAF数据集外加一个合成任务上评估AFR。五次种子重复验证显示,在四个FLamby任务和六个LEAF配置中获得显著提升;同时揭示了对种子敏感和仅单次运行有效的失败案例。相同预算对照表明,部分增益源于算法配方变更,其他改进可由固定表面标量调优实现,或在重复与保留测试中失效。这些混合结果正是贡献所在:揭示了代理生成候选方案可区分为可重复的联邦机制、固定表面调优效应与特定单次运行产物。

原文摘要 · Abstract (English)

Federated learning (FL) research often depends on many small but consequential algorithmic choices: optimizer variants, server aggregation rules, local training schedules, normalization, regularization, and model architecture. These choices are expensive to explore manually and difficult to compare fairly when candidate changes can also alter the FL training or evaluation path. In this work, we present Auto-FL-Research (AFR), a constrained coding-agent workflow for FL algorithmic recipe search. Agents may propose and implement candidate training algorithms, including server aggregation rules, client update schedules, local objectives, and registered model variants, while task profiles fix the mutation surface, compute budget, communication contract, and final model evaluation. Each campaign records candidate scores, runtime, edited files, artifacts, and failure status. We evaluate AFR on five healthcare cross-silo FLamby tasks and on grouped-client profiles for the five fixed LEAF datasets plus the LEAF synthetic task. Five-seed repeat evaluations support gains on four FLamby tasks and five of six LEAF profiles, while also exposing seed-sensitive and search-selected failure cases. Same-budget controls show that several gains correspond to FL-recipe changes, whereas other improvements are recovered by fixed-surface scalar controls or fail under repeat or held-out evaluation. These mixed outcomes are part of the contribution: they show how agent-generated candidates can be separated into repeated FL mechanisms, fixed-surface tuning effects, and selected single-run artifacts.

联邦学习智能体算法搜索医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。