跳到正文
OpenReview · Representation learning· Jingrong Xie; Yuqi Feng; Minxiao Zhong; Yanan Sun·· 2025-01-01精选AI 评分75

基于对比表征学习的神经网络架构生成

Neural Architecture Generation via Contrastive Representation Learning

AI 导读

基于摘要分析。该研究针对神经网络架构搜索需要逐一评估大量候选架构、时间与计算开销较高的问题,提出结合对比表征学习与自引导扩散模型的架构生成方法。其主要研究对象是通用神经网络架构设计,而非 EEG 解码或 BCI。 机制上,对比学习用于同时捕捉架构的静态属性与潜在性能特性,自引导扩散模型随后学习这些表征,并面向不同任务生成架构。作者主张该方法可避免生成过程中的迭代评估与额外训练;但这一表述是否涉及表征学习、性能监督及扩散模型的前期训练成本,仍需全文核对。具体架构编码、对比样本构造、损失函数、引导机制及生成后的有效性约束尚未验证。 作者报告,架构生成仅需 0.02 GPU Days,并给出所生成架构在 ImagNet(摘要原文拼写)上达到 77.1% top-1 Accuracy、在 CIFAR-10 上达到 97.65% Accuracy 的结果。摘要未明确两项结果对应的完整训练与评估协议,也未说明生成预算是否包含前期训练和最终架构训练,不能将其视为完整端到端成本。作者还报告可在 DARTS、NAS-Bench-201 和 AutoFormer 等卷积及 Transformer 搜索空间中找到具有竞争力的架构;具体对照、数据划分、消融及统计信息尚未验证。 平台推测(待验证):若性能相关架构表征能够在低成本监督下学习,该机制可能用于减少 EEG 架构搜索中的重复训练。可复用的是表征学习与生成流程,需改造的是适配时序信号和通道结构的搜索空间、性能监督与任务条件。低信噪比、通道差异、小数据及跨被试性能波动可能使生成器偏向不稳健的架构。一个可检验的小实验是在固定 EEG 数据划分与统一训练预算下,对比生成架构和随机搜索的 Cross-subject Accuracy、计算成本及多次运行波动,并单独计入生成器前期训练成本。已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对其对比式架构表征与自引导扩散生成的结合机制,尤其是生成成本与完整搜索成本的统计边界;其 EEG/BCI 适用性尚未验证。

来源:OpenReview · Representation learning · openreview.net