Skip to content

[Stage C1] VisDrone MoE 武器横评:P2 检测头 + UoMoE 变体公平同预算对比(50 epoch, imgsz=640) - #255

Open
OldPigxjk wants to merge 3 commits into
Tencent:mainfrom
OldPigxjk:c1-submit
Open

[Stage C1] VisDrone MoE 武器横评:P2 检测头 + UoMoE 变体公平同预算对比(50 epoch, imgsz=640)#255
OldPigxjk wants to merge 3 commits into
Tencent:mainfrom
OldPigxjk:c1-submit

Conversation

@OldPigxjk

Copy link
Copy Markdown

第三阶段实战开发 C1:VisDrone MoE 武器横评

背景

腾讯犀牛鸟开源课题第三阶段实战开发(YOLO-Master)C1 任务交付:在 VisDrone(航拍密集小目标)上对 MoE 变体做公平同预算横评。本工作基于 Issue #49(v0.1-N vs EsMoE-N 基线复现)扩展,引入 P2/4 检测头与 UltraOptimizedMoE 变体。

导师方案确认(2026-09-01)

  1. 分辨率可降到 640:本实验统一采用 imgsz=640(覆盖 P2 检测头在 8GB 显存下的可行性)。
  2. 此前跑出的结果可作为基线:5-epoch smoke(mAP50≈0.05)+ 50-epoch 基线(v0.1-N=0.1959 / EsMoE-N=0.19041)均作为正式基线。
  3. COCO APs 口径正确:small = 实例面积 < 1024,maxDets = 500(标准 COCO 定义)。

训练配置(公平同预算)

  • 4 武器:EsMoE-P2-N / v0.1-P2-N / UoMoE-N / UoMoE-P2-N
  • epochs=50, batch=8, imgsz=640, device=0 (RTX 4060 8GB), workers=0, seed=42
  • deterministic=True, pretrained=False, lora_r=0, optimizer=auto(SGD@0.01), amp=False
  • --no-sparse-eval(dense eval 修正 ES-MoE sparse 塌陷,train==eval 一致)
  • 与 Stage1 基线严格同预算对比

结果(VisDrone, 50 epoch, mAP50 / mAP50-95)

武器 mAP50 mAP50-95
EsMoE-P2-N 0.20773 0.11405
v0.1-P2-N 0.20160 0.11081
UoMoE-N 0.19626 0.10570
UoMoE-P2-N 0.20503 0.11337

基线对照(c1_onoff):v0.1-N=0.19590 / EsMoE-N=0.19041(mAP50)
三 seed 稳健性(EsMoE-P2-N):seed42=0.20773 / seed0=0.19990 / seed1=0.19673(mAP50)

交付物

  • 复现脚本:smoke/c1/run_c1_weapons.py、run_c1_baseline.py、smoke_c1_esmoe_n.py
  • 配置:smoke/c1/VisDrone_local.yaml、requirements_smoke.txt、audit_params.py
  • 训练日志与汇总:各 runs/c1_*/summary.csv(逐 epoch 指标见 results.csv)
  • 说明文档:smoke/c1/README.md、RESULTS_stage1.md、STAGE2_STATUS.md

已知限制(诚实标注)

  • COCO APs 评估:EsMoE-P2-N 已生成 aps_result.json,其余武器为后续补充项
  • SKU-110K 数据集未纳入本轮(资源优先 VisDrone 密集小目标消融)
  • 完整 VisDrone 数据集 50 epoch,未在 100~300 epoch 区间进一步拉满(受 RTX 4060 8GB 单卡预算约束)

OldPigxjk and others added 3 commits August 27, 2026 22:08
…cce839c)

Submit branch based on 2026-08-21 main baseline acce839 per 增量验收补充细则.
- run_c1_baseline.py / run_c1_weapons.py: on/off + 4-weapon横评, best.pt lock-safe
- audit_params.py: 6-weapon param self-audit (nano compliant)
- requirements_smoke.txt: pinned deps
- RESULTS_stage1.md / STAGE2_STATUS.md: experiment evidence
…iner.save_model: best.pt lock skip; last.pt 5x retry - _save_run_args: args.yaml lock skip - Stage2 EsMoE-P2-N done; v0.1-P2-N resuming
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants