dsh-aris-panel 0.2.7 → 0.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +117 -108
- package/README_CN.md +98 -89
- package/aris/.claude-plugin/marketplace.json +34 -0
- package/aris/.claude-plugin/plugin.json +19 -0
- package/aris/.codex-plugin/plugin.json +18 -0
- package/aris/.env.example +53 -0
- package/aris/.gitattributes +6 -0
- package/aris/.github/agents/aris-reviewer-claude.agent.md +10 -0
- package/aris/.github/agents/aris-reviewer-openai.agent.md +10 -0
- package/aris/.github/workflows/check-skills-inventory.yml +32 -0
- package/aris/.github/workflows/lint-skills-helpers.yml +38 -0
- package/aris/.github/workflows/test-install-aris-ps1.yml +31 -0
- package/aris/.github/workflows/test-suite.yml +59 -0
- package/aris/AGENT_GUIDE.md +222 -0
- package/aris/CONTRIBUTING.md +83 -0
- package/aris/CONTRIBUTING_CN.md +83 -0
- package/aris/LICENSE +21 -0
- package/aris/README.md +1739 -0
- package/aris/README_CN.md +1526 -0
- package/aris/SETUP_GUIDE.md +279 -0
- package/aris/SETUP_GUIDE_CN.md +279 -0
- package/aris/UPSTREAM.txt +2 -0
- package/aris/aris-monitor/README.md +158 -0
- package/aris/aris-monitor/assets/screenshot.png +0 -0
- package/aris/aris-monitor/focus-tty.sh +156 -0
- package/aris/aris-monitor/focus.py +93 -0
- package/aris/aris-monitor/run.sh +49 -0
- package/aris/aris-monitor/scanner.py +445 -0
- package/aris/aris-monitor/ticker.py +81 -0
- package/aris/aris-monitor/widget.py +372 -0
- package/aris/assets/claude-fleet-preview.png +0 -0
- package/aris/assets/community_showcase_7_10_codex.png +0 -0
- package/aris/assets/community_showcase_8_10.png +0 -0
- package/aris/assets/feishu_interactive.jpg +0 -0
- package/aris/assets/feishu_push.png +0 -0
- package/aris/assets/paper_illustration_demo.png +0 -0
- package/aris/commands/setup.md +77 -0
- package/aris/community_papers/UAV-CC.pdf +0 -0
- package/aris/docs/ALI_CODING_PLAN_GUIDE.md +324 -0
- package/aris/docs/ANTIGRAVITY_ADAPTATION.md +409 -0
- package/aris/docs/ANTIGRAVITY_ADAPTATION_CN.md +363 -0
- package/aris/docs/ARIS-Code-README_CN.md +8 -0
- package/aris/docs/ARIS-Code-README_EN.md +8 -0
- package/aris/docs/ARIS_INTRO.html +902 -0
- package/aris/docs/ARIS_INTRO.md +368 -0
- package/aris/docs/ARIS_INTRO.review.json +79 -0
- package/aris/docs/CODEX_CLAUDE_REVIEW_GUIDE.md +116 -0
- package/aris/docs/CODEX_CLAUDE_REVIEW_GUIDE_CN.md +116 -0
- package/aris/docs/CODEX_GEMINI_REVIEW_GUIDE.md +249 -0
- package/aris/docs/CODEX_GEMINI_REVIEW_GUIDE_CN.md +250 -0
- package/aris/docs/COPILOT_CLI_ADAPTATION.md +388 -0
- package/aris/docs/CURSOR_ADAPTATION.md +305 -0
- package/aris/docs/CUSTOMIZATION.md +155 -0
- package/aris/docs/CUSTOMIZATION_CN.md +154 -0
- package/aris/docs/GPU_SETUP.md +57 -0
- package/aris/docs/GPU_SETUP_CN.md +54 -0
- package/aris/docs/LLM_API_MIX_MATCH_GUIDE.md +256 -0
- package/aris/docs/MANUAL_REVIEW_GUIDE.md +128 -0
- package/aris/docs/MANUAL_REVIEW_GUIDE_CN.md +116 -0
- package/aris/docs/MINIMAX_MCP_GUIDE.md +198 -0
- package/aris/docs/MODELSCOPE_GUIDE.md +305 -0
- package/aris/docs/MODEL_COMBINATIONS.md +134 -0
- package/aris/docs/MODEL_COMBINATIONS_CN.md +134 -0
- package/aris/docs/MiniMax-GLM-Configuration.md +195 -0
- package/aris/docs/NARRATIVE_REPORT_EXAMPLE.md +143 -0
- package/aris/docs/OPENCLAW_ADAPTATION.md +59 -0
- package/aris/docs/OPENROUTER_GUIDE.md +255 -0
- package/aris/docs/OPENROUTER_GUIDE_CN.md +255 -0
- package/aris/docs/PROJECT_FILES_GUIDE.md +161 -0
- package/aris/docs/PROJECT_FILES_GUIDE_CN.md +161 -0
- package/aris/docs/RESUBMIT_AND_TALK.md +91 -0
- package/aris/docs/RESUBMIT_AND_TALK_CN.md +91 -0
- package/aris/docs/SESSION_RECOVERY_GUIDE.md +439 -0
- package/aris/docs/SESSION_RECOVERY_GUIDE_CN.md +431 -0
- package/aris/docs/SKILLS_CATALOG.md +204 -0
- package/aris/docs/TRAE_ARIS_RUNBOOK_CN.md +329 -0
- package/aris/docs/TRAE_ARIS_RUNBOOK_EN.md +329 -0
- package/aris/docs/WATCHDOG_GUIDE.md +200 -0
- package/aris/docs/WATCHDOG_GUIDE_CN.md +199 -0
- package/aris/docs/aris-code-banner.png +0 -0
- package/aris/docs/aris-code-screenshot.png +0 -0
- package/aris/docs/aris-movie-director-method.png +0 -0
- package/aris/docs/aris_intro_slides.pdf +0 -0
- package/aris/docs/aris_logo.svg +126 -0
- package/aris/docs/auto_review_score_curve.png +0 -0
- package/aris/docs/blogs/continuous_dlm_2026h1_survey.html +3262 -0
- package/aris/docs/hero_combined.svg +844 -0
- package/aris/docs/hero_idea.svg +158 -0
- package/aris/docs/hero_implement.svg +109 -0
- package/aris/docs/hero_interaction.svg +172 -0
- package/aris/docs/hf_daily_paper_1.svg +19 -0
- package/aris/docs/integrations/FEISHU.md +208 -0
- package/aris/docs/integrations/FEISHU_CN.md +208 -0
- package/aris/docs/integrations/OBSIDIAN.md +65 -0
- package/aris/docs/integrations/OBSIDIAN_CN.md +65 -0
- package/aris/docs/integrations/VAST_GPU_GUIDE.md +96 -0
- package/aris/docs/integrations/VAST_GPU_GUIDE_CN.md +96 -0
- package/aris/docs/integrations/ZOTERO.md +57 -0
- package/aris/docs/integrations/ZOTERO_CN.md +57 -0
- package/aris/docs/tutorials/3d_generation_tutorial.html +1018 -0
- package/aris/docs/tutorials/3d_generation_tutorial.md +1206 -0
- package/aris/docs/tutorials/3d_generation_tutorial.review.json +34 -0
- package/aris/docs/tutorials/3d_generation_tutorial_en.html +1018 -0
- package/aris/docs/tutorials/3d_generation_tutorial_en.md +1206 -0
- package/aris/docs/tutorials/README.md +77 -0
- package/aris/docs/tutorials/agent_foundations_tutorial.html +1081 -0
- package/aris/docs/tutorials/agent_foundations_tutorial.md +1229 -0
- package/aris/docs/tutorials/agent_foundations_tutorial.review.json +146 -0
- package/aris/docs/tutorials/agent_foundations_tutorial_en.html +1081 -0
- package/aris/docs/tutorials/agent_foundations_tutorial_en.md +1229 -0
- package/aris/docs/tutorials/agentic_rl_tutorial.html +1269 -0
- package/aris/docs/tutorials/agentic_rl_tutorial.md +1544 -0
- package/aris/docs/tutorials/agentic_rl_tutorial.review.json +74 -0
- package/aris/docs/tutorials/agentic_rl_tutorial_en.html +1269 -0
- package/aris/docs/tutorials/agentic_rl_tutorial_en.md +1544 -0
- package/aris/docs/tutorials/attention_tutorial.html +774 -0
- package/aris/docs/tutorials/attention_tutorial.md +812 -0
- package/aris/docs/tutorials/attention_tutorial.review.json +51 -0
- package/aris/docs/tutorials/attention_tutorial_en.html +774 -0
- package/aris/docs/tutorials/attention_tutorial_en.md +812 -0
- package/aris/docs/tutorials/diffusion_distillation_tutorial.html +1421 -0
- package/aris/docs/tutorials/diffusion_distillation_tutorial.md +1672 -0
- package/aris/docs/tutorials/diffusion_distillation_tutorial.review.json +114 -0
- package/aris/docs/tutorials/diffusion_distillation_tutorial_en.html +1421 -0
- package/aris/docs/tutorials/diffusion_distillation_tutorial_en.md +1672 -0
- package/aris/docs/tutorials/diffusion_foundations_tutorial.html +1239 -0
- package/aris/docs/tutorials/diffusion_foundations_tutorial.md +1481 -0
- package/aris/docs/tutorials/diffusion_foundations_tutorial.review.json +56 -0
- package/aris/docs/tutorials/diffusion_foundations_tutorial_en.html +1239 -0
- package/aris/docs/tutorials/diffusion_foundations_tutorial_en.md +1481 -0
- package/aris/docs/tutorials/diffusion_post_training_tutorial.html +1135 -0
- package/aris/docs/tutorials/diffusion_post_training_tutorial.md +1322 -0
- package/aris/docs/tutorials/diffusion_post_training_tutorial.review.json +63 -0
- package/aris/docs/tutorials/diffusion_post_training_tutorial_en.html +1138 -0
- package/aris/docs/tutorials/diffusion_post_training_tutorial_en.md +1325 -0
- package/aris/docs/tutorials/distributed_training_tutorial.html +1429 -0
- package/aris/docs/tutorials/distributed_training_tutorial.md +1587 -0
- package/aris/docs/tutorials/distributed_training_tutorial.review.json +98 -0
- package/aris/docs/tutorials/distributed_training_tutorial_en.html +1429 -0
- package/aris/docs/tutorials/distributed_training_tutorial_en.md +1587 -0
- package/aris/docs/tutorials/flow_matching_tutorial.html +755 -0
- package/aris/docs/tutorials/flow_matching_tutorial.md +537 -0
- package/aris/docs/tutorials/flow_matching_tutorial.review.json +37 -0
- package/aris/docs/tutorials/flow_matching_tutorial_en.html +755 -0
- package/aris/docs/tutorials/flow_matching_tutorial_en.md +537 -0
- package/aris/docs/tutorials/image_generation_systems_tutorial.html +1181 -0
- package/aris/docs/tutorials/image_generation_systems_tutorial.md +1448 -0
- package/aris/docs/tutorials/image_generation_systems_tutorial.review.json +49 -0
- package/aris/docs/tutorials/image_generation_systems_tutorial_en.html +1181 -0
- package/aris/docs/tutorials/image_generation_systems_tutorial_en.md +1448 -0
- package/aris/docs/tutorials/kl_divergence_rlhf_tutorial.html +1210 -0
- package/aris/docs/tutorials/kl_divergence_rlhf_tutorial.md +1502 -0
- package/aris/docs/tutorials/kl_divergence_rlhf_tutorial.review.json +55 -0
- package/aris/docs/tutorials/kl_divergence_rlhf_tutorial_en.html +1212 -0
- package/aris/docs/tutorials/kl_divergence_rlhf_tutorial_en.md +1504 -0
- package/aris/docs/tutorials/kv_cache_speculative_decoding_tutorial.html +1118 -0
- package/aris/docs/tutorials/kv_cache_speculative_decoding_tutorial.md +1258 -0
- package/aris/docs/tutorials/kv_cache_speculative_decoding_tutorial.review.json +91 -0
- package/aris/docs/tutorials/kv_cache_speculative_decoding_tutorial_en.html +1118 -0
- package/aris/docs/tutorials/kv_cache_speculative_decoding_tutorial_en.md +1258 -0
- package/aris/docs/tutorials/llm_opd_tutorial.html +1109 -0
- package/aris/docs/tutorials/llm_opd_tutorial.md +1195 -0
- package/aris/docs/tutorials/llm_opd_tutorial.review.json +125 -0
- package/aris/docs/tutorials/llm_opd_tutorial_en.html +1112 -0
- package/aris/docs/tutorials/llm_opd_tutorial_en.md +1198 -0
- package/aris/docs/tutorials/long_context_rope_yarn_mla_tutorial.html +1130 -0
- package/aris/docs/tutorials/long_context_rope_yarn_mla_tutorial.md +1209 -0
- package/aris/docs/tutorials/long_context_rope_yarn_mla_tutorial.review.json +79 -0
- package/aris/docs/tutorials/long_context_rope_yarn_mla_tutorial_en.html +1132 -0
- package/aris/docs/tutorials/long_context_rope_yarn_mla_tutorial_en.md +1211 -0
- package/aris/docs/tutorials/moe_tutorial.html +968 -0
- package/aris/docs/tutorials/moe_tutorial.md +1031 -0
- package/aris/docs/tutorials/moe_tutorial.review.json +54 -0
- package/aris/docs/tutorials/moe_tutorial_en.html +968 -0
- package/aris/docs/tutorials/moe_tutorial_en.md +1031 -0
- package/aris/docs/tutorials/multi_agent_long_horizon_tutorial.html +1409 -0
- package/aris/docs/tutorials/multi_agent_long_horizon_tutorial.md +1803 -0
- package/aris/docs/tutorials/multi_agent_long_horizon_tutorial.review.json +62 -0
- package/aris/docs/tutorials/multi_agent_long_horizon_tutorial_en.html +1413 -0
- package/aris/docs/tutorials/multi_agent_long_horizon_tutorial_en.md +1807 -0
- package/aris/docs/tutorials/quantization_tutorial.html +1117 -0
- package/aris/docs/tutorials/quantization_tutorial.md +1344 -0
- package/aris/docs/tutorials/quantization_tutorial.review.json +56 -0
- package/aris/docs/tutorials/quantization_tutorial_en.html +1117 -0
- package/aris/docs/tutorials/quantization_tutorial_en.md +1344 -0
- package/aris/docs/tutorials/reasoning_models_tutorial.html +1085 -0
- package/aris/docs/tutorials/reasoning_models_tutorial.md +1285 -0
- package/aris/docs/tutorials/reasoning_models_tutorial.review.json +66 -0
- package/aris/docs/tutorials/reasoning_models_tutorial_en.html +1085 -0
- package/aris/docs/tutorials/reasoning_models_tutorial_en.md +1285 -0
- package/aris/docs/tutorials/rlhf_dpo_grpo_ppo_tutorial.html +1015 -0
- package/aris/docs/tutorials/rlhf_dpo_grpo_ppo_tutorial.md +1125 -0
- package/aris/docs/tutorials/rlhf_dpo_grpo_ppo_tutorial.review.json +69 -0
- package/aris/docs/tutorials/rlhf_dpo_grpo_ppo_tutorial_en.html +1014 -0
- package/aris/docs/tutorials/rlhf_dpo_grpo_ppo_tutorial_en.md +1123 -0
- package/aris/docs/tutorials/self_evolving_agents_tutorial.html +1354 -0
- package/aris/docs/tutorials/self_evolving_agents_tutorial.md +1554 -0
- package/aris/docs/tutorials/self_evolving_agents_tutorial.review.json +64 -0
- package/aris/docs/tutorials/self_evolving_agents_tutorial_en.html +1356 -0
- package/aris/docs/tutorials/self_evolving_agents_tutorial_en.md +1556 -0
- package/aris/docs/tutorials/vae_vqvae_vqgan_tutorial.html +1197 -0
- package/aris/docs/tutorials/vae_vqvae_vqgan_tutorial.md +1419 -0
- package/aris/docs/tutorials/vae_vqvae_vqgan_tutorial.review.json +67 -0
- package/aris/docs/tutorials/vae_vqvae_vqgan_tutorial_en.html +1196 -0
- package/aris/docs/tutorials/vae_vqvae_vqgan_tutorial_en.md +1419 -0
- package/aris/docs/tutorials/video_generation_tutorial.html +1169 -0
- package/aris/docs/tutorials/video_generation_tutorial.md +1189 -0
- package/aris/docs/tutorials/video_generation_tutorial.review.json +72 -0
- package/aris/docs/tutorials/video_generation_tutorial_en.html +1170 -0
- package/aris/docs/tutorials/video_generation_tutorial_en.md +1190 -0
- package/aris/docs/tutorials/vlm_multimodal_tutorial.html +1208 -0
- package/aris/docs/tutorials/vlm_multimodal_tutorial.md +1446 -0
- package/aris/docs/tutorials/vlm_multimodal_tutorial.review.json +67 -0
- package/aris/docs/tutorials/vlm_multimodal_tutorial_en.html +1208 -0
- package/aris/docs/tutorials/vlm_multimodal_tutorial_en.md +1446 -0
- package/aris/docs/wechat_group.jpg +0 -0
- package/aris/mcp-servers/claude-review/README.md +96 -0
- package/aris/mcp-servers/claude-review/run_with_claude_aws.sh +49 -0
- package/aris/mcp-servers/claude-review/server.py +718 -0
- package/aris/mcp-servers/codex-exec/README.md +57 -0
- package/aris/mcp-servers/codex-exec/server.py +480 -0
- package/aris/mcp-servers/codex-image2/README.md +65 -0
- package/aris/mcp-servers/codex-image2/server.py +893 -0
- package/aris/mcp-servers/feishu-bridge/requirements.txt +1 -0
- package/aris/mcp-servers/feishu-bridge/server.py +240 -0
- package/aris/mcp-servers/gemini-review/README.md +171 -0
- package/aris/mcp-servers/gemini-review/server.py +1856 -0
- package/aris/mcp-servers/llm-chat/requirements.txt +1 -0
- package/aris/mcp-servers/llm-chat/server.py +664 -0
- package/aris/mcp-servers/manual-review/README.md +133 -0
- package/aris/mcp-servers/manual-review/server.py +910 -0
- package/aris/mcp-servers/manual-review/ui.html +279 -0
- package/aris/mcp-servers/minimax-chat/requirements.txt +1 -0
- package/aris/mcp-servers/minimax-chat/server.py +381 -0
- package/aris/skills/ablation-planner/SKILL.md +123 -0
- package/aris/skills/alphaxiv/SKILL.md +196 -0
- package/aris/skills/analyze-results/SKILL.md +46 -0
- package/aris/skills/arxiv/SKILL.md +248 -0
- package/aris/skills/auto-paper-improvement-loop/SKILL.md +695 -0
- package/aris/skills/auto-review-loop/SKILL.md +1137 -0
- package/aris/skills/auto-review-loop-llm/SKILL.md +259 -0
- package/aris/skills/auto-review-loop-minimax/SKILL.md +302 -0
- package/{skills/citation-audit-claude → aris/skills/citation-audit}/SKILL.md +11 -21
- package/aris/skills/claims-drafting/SKILL.md +227 -0
- package/aris/skills/comm-lit-review/SKILL.md +297 -0
- package/aris/skills/deepxiv/SKILL.md +263 -0
- package/aris/skills/dse-loop/SKILL.md +296 -0
- package/aris/skills/embodiment-description/SKILL.md +129 -0
- package/aris/skills/exa-search/SKILL.md +205 -0
- package/{skills/experiment-audit-claude → aris/skills/experiment-audit}/SKILL.md +16 -26
- package/aris/skills/experiment-bridge/SKILL.md +376 -0
- package/aris/skills/experiment-plan/SKILL.md +249 -0
- package/aris/skills/experiment-queue/SKILL.md +431 -0
- package/aris/skills/experiment-queue/scripts/build_manifest.py +142 -0
- package/aris/skills/experiment-queue/scripts/queue_manager.py +433 -0
- package/aris/skills/feishu-notify/SKILL.md +156 -0
- package/aris/skills/figure-description/SKILL.md +138 -0
- package/aris/skills/figure-spec/SKILL.md +262 -0
- package/aris/skills/figure-spec/scripts/figure_renderer.py +799 -0
- package/aris/skills/formula-derivation/SKILL.md +280 -0
- package/aris/skills/gemini-search/SKILL.md +231 -0
- package/aris/skills/grant-proposal/SKILL.md +698 -0
- package/aris/skills/idea-creator/SKILL.md +584 -0
- package/aris/skills/idea-discovery/SKILL.md +527 -0
- package/aris/skills/idea-discovery-robot/SKILL.md +363 -0
- package/aris/skills/integrity-forensics/SKILL.md +284 -0
- package/aris/skills/interview-cheatsheet/SKILL.md +245 -0
- package/aris/skills/invention-structuring/SKILL.md +188 -0
- package/aris/skills/jurisdiction-format/SKILL.md +192 -0
- package/aris/skills/kill-argument/SKILL.md +437 -0
- package/aris/skills/mermaid-diagram/SKILL.md +419 -0
- package/{skills/meta-apply-claude → aris/skills/meta-apply}/SKILL.md +5 -15
- package/aris/skills/meta-optimize/SKILL.md +437 -0
- package/aris/skills/monitor-experiment/SKILL.md +140 -0
- package/aris/skills/novelty-check/SKILL.md +142 -0
- package/aris/skills/openalex/SKILL.md +237 -0
- package/aris/skills/overleaf-sync/SKILL.md +220 -0
- package/{skills/paper-claim-audit-claude → aris/skills/paper-claim-audit}/SKILL.md +13 -23
- package/aris/skills/paper-compile/SKILL.md +266 -0
- package/aris/skills/paper-figure/SKILL.md +312 -0
- package/aris/skills/paper-illustration/SKILL.md +736 -0
- package/aris/skills/paper-illustration-image2/SKILL.md +391 -0
- package/aris/skills/paper-illustration-image2/scripts/paper_illustration_image2.py +255 -0
- package/aris/skills/paper-plan/SKILL.md +386 -0
- package/aris/skills/paper-poster/SKILL.md +19 -0
- package/aris/skills/paper-poster-html/DESIGN_FINAL.md +176 -0
- package/aris/skills/paper-poster-html/IMPLEMENTATION_CONVENTIONS.md +161 -0
- package/aris/skills/paper-poster-html/LICENSES/posterly-MIT.txt +21 -0
- package/aris/skills/paper-poster-html/NOTICE.md +57 -0
- package/aris/skills/paper-poster-html/SKILL.md +323 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/__init__.py +0 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/canvas.py +200 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/measure.py +588 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/polish.py +498 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/preflight.py +489 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/render.py +215 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/textutil.py +16 -0
- package/aris/skills/paper-poster-html/scripts/_posterly/verify_final.py +171 -0
- package/aris/skills/paper-poster-html/scripts/asset_check.py +897 -0
- package/aris/skills/paper-poster-html/scripts/extract_pdf_figures.py +666 -0
- package/aris/skills/paper-poster-html/scripts/poster_check.py +251 -0
- package/aris/skills/paper-poster-html/scripts/preprocess_figures.py +238 -0
- package/aris/skills/paper-poster-html/scripts/render_preview.py +217 -0
- package/aris/skills/paper-poster-html/scripts/run_gates.py +556 -0
- package/aris/skills/paper-poster-html/scripts/style_check.py +1324 -0
- package/aris/skills/paper-poster-html/templates/COMPONENTS.md +462 -0
- package/aris/skills/paper-poster-html/templates/README.md +170 -0
- package/aris/skills/paper-poster-html/templates/landscape_4col.html +1032 -0
- package/aris/skills/paper-poster-html/templates/landscape_hero.html +1046 -0
- package/aris/skills/paper-poster-html/templates/portrait_2col.html +947 -0
- package/aris/skills/paper-poster-html/templates/tokens/acl.json +9 -0
- package/aris/skills/paper-poster-html/templates/tokens/cvpr.json +9 -0
- package/aris/skills/paper-poster-html/templates/tokens/generic.json +9 -0
- package/aris/skills/paper-poster-html/templates/tokens/iclr.json +9 -0
- package/aris/skills/paper-poster-html/templates/tokens/icml.json +9 -0
- package/aris/skills/paper-poster-html/templates/tokens/neurips.json +9 -0
- package/aris/skills/paper-slides/SKILL.md +635 -0
- package/aris/skills/paper-talk/SKILL.md +381 -0
- package/aris/skills/paper-write/SKILL.md +682 -0
- package/aris/skills/paper-write/templates/IEEEtran.bst +2409 -0
- package/aris/skills/paper-write/templates/IEEEtran.cls +6347 -0
- package/aris/skills/paper-write/templates/iclr2026.tex +84 -0
- package/aris/skills/paper-write/templates/icml2025.tex +87 -0
- package/aris/skills/paper-write/templates/ieee_conference.tex +89 -0
- package/aris/skills/paper-write/templates/ieee_journal.tex +93 -0
- package/aris/skills/paper-write/templates/math_commands.tex +48 -0
- package/aris/skills/paper-write/templates/neurips2025.tex +80 -0
- package/aris/skills/paper-writing/SKILL.md +929 -0
- package/aris/skills/patent-novelty-check/SKILL.md +153 -0
- package/aris/skills/patent-pipeline/SKILL.md +344 -0
- package/aris/skills/patent-review/SKILL.md +203 -0
- package/aris/skills/pixel-art/SKILL.md +137 -0
- package/aris/skills/prior-art-search/SKILL.md +146 -0
- package/aris/skills/proof-checker/SKILL.md +866 -0
- package/aris/skills/proof-orchestrator/NOTICE.md +24 -0
- package/aris/skills/proof-orchestrator/SKILL.md +254 -0
- package/aris/skills/proof-orchestrator/references/audit-output-contract.md +126 -0
- package/aris/skills/proof-orchestrator/references/deepseek-routing.md +74 -0
- package/aris/skills/proof-orchestrator/references/dispatch-prompts.md +227 -0
- package/aris/skills/proof-orchestrator/references/notation-audit.md +135 -0
- package/aris/skills/proof-orchestrator/references/proof-audit-rubric.md +70 -0
- package/aris/skills/proof-orchestrator/references/stress-tests.md +38 -0
- package/aris/skills/proof-writer/SKILL.md +223 -0
- package/aris/skills/qzcli/SKILL.md +324 -0
- package/aris/skills/rebuttal/SKILL.md +376 -0
- package/aris/skills/render-html/SKILL.md +316 -0
- package/aris/skills/render-html/scripts/render_html.py +1006 -0
- package/aris/skills/render-html/scripts/templates/academic.html +703 -0
- package/aris/skills/render-html/scripts/templates/dashboard.html +333 -0
- package/aris/skills/research-implement-feature/SKILL.md +583 -0
- package/aris/skills/research-lit/SKILL.md +756 -0
- package/aris/skills/research-pipeline/SKILL.md +390 -0
- package/aris/skills/research-refine/SKILL.md +770 -0
- package/aris/skills/research-refine-pipeline/SKILL.md +186 -0
- package/aris/skills/research-review/SKILL.md +223 -0
- package/aris/skills/research-wiki/SKILL.md +461 -0
- package/aris/skills/resubmit-pipeline/SKILL.md +447 -0
- package/{skills/result-to-claim-claude → aris/skills/result-to-claim}/SKILL.md +7 -17
- package/aris/skills/run-experiment/SKILL.md +313 -0
- package/aris/skills/semantic-scholar/SKILL.md +236 -0
- package/aris/skills/serverless-modal/SKILL.md +335 -0
- package/aris/skills/shared-references/acceptance-gate.md +324 -0
- package/aris/skills/shared-references/assurance-contract.md +248 -0
- package/aris/skills/shared-references/capture-antipatterns.md +78 -0
- package/aris/skills/shared-references/citation-discipline.md +583 -0
- package/aris/skills/shared-references/compute-env-contract.md +163 -0
- package/aris/skills/shared-references/effort-contract.md +183 -0
- package/aris/skills/shared-references/evidence-precheck.md +65 -0
- package/aris/skills/shared-references/experiment-integrity.md +49 -0
- package/aris/skills/shared-references/external-cadence.md +326 -0
- package/aris/skills/shared-references/fan-out-pattern.md +366 -0
- package/aris/skills/shared-references/injection-hygiene.md +127 -0
- package/aris/skills/shared-references/integration-contract.md +461 -0
- package/aris/skills/shared-references/output-composition.md +93 -0
- package/aris/skills/shared-references/output-language.md +45 -0
- package/aris/skills/shared-references/output-manifest.md +49 -0
- package/aris/skills/shared-references/output-versioning.md +111 -0
- package/aris/skills/shared-references/patent-format-cn.md +199 -0
- package/aris/skills/shared-references/patent-format-ep.md +173 -0
- package/aris/skills/shared-references/patent-format-us.md +161 -0
- package/aris/skills/shared-references/patent-writing-principles.md +197 -0
- package/aris/skills/shared-references/prior-art-databases.md +141 -0
- package/aris/skills/shared-references/resumable-runs.md +109 -0
- package/aris/skills/shared-references/review-scope-limits.md +81 -0
- package/aris/skills/shared-references/review-tracing.md +391 -0
- package/aris/skills/shared-references/reviewer-independence.md +79 -0
- package/aris/skills/shared-references/reviewer-routing.md +852 -0
- package/aris/skills/shared-references/skill-governance.md +104 -0
- package/aris/skills/shared-references/taste-calibration.md +85 -0
- package/aris/skills/shared-references/venue-checklists.md +114 -0
- package/aris/skills/shared-references/wiki-helper-resolution.md +134 -0
- package/aris/skills/shared-references/writing-principles.md +525 -0
- package/aris/skills/skills-codex/README.md +102 -0
- package/aris/skills/skills-codex/README_CN.md +100 -0
- package/aris/skills/skills-codex/ablation-planner/SKILL.md +126 -0
- package/aris/skills/skills-codex/alphaxiv/SKILL.md +186 -0
- package/aris/skills/skills-codex/analyze-results/SKILL.md +45 -0
- package/aris/skills/skills-codex/arxiv/SKILL.md +210 -0
- package/aris/skills/skills-codex/auto-paper-improvement-loop/SKILL.md +618 -0
- package/aris/skills/skills-codex/auto-review-loop/SKILL.md +500 -0
- package/aris/skills/skills-codex/auto-review-loop-llm/SKILL.md +247 -0
- package/aris/skills/skills-codex/auto-review-loop-minimax/SKILL.md +290 -0
- package/aris/skills/skills-codex/citation-audit/SKILL.md +504 -0
- package/aris/skills/skills-codex/claims-drafting/SKILL.md +239 -0
- package/aris/skills/skills-codex/comm-lit-review/SKILL.md +299 -0
- package/aris/skills/skills-codex/comm-lit-review/references/domain-taxonomy.md +57 -0
- package/aris/skills/skills-codex/comm-lit-review/references/output-template.md +37 -0
- package/aris/skills/skills-codex/comm-lit-review/references/source-policy.md +99 -0
- package/aris/skills/skills-codex/comm-lit-review/references/venue-tiering.md +112 -0
- package/aris/skills/skills-codex/deepxiv/SKILL.md +142 -0
- package/aris/skills/skills-codex/dse-loop/SKILL.md +285 -0
- package/aris/skills/skills-codex/embodiment-description/SKILL.md +129 -0
- package/aris/skills/skills-codex/exa-search/SKILL.md +192 -0
- package/aris/skills/skills-codex/experiment-audit/SKILL.md +286 -0
- package/aris/skills/skills-codex/experiment-bridge/SKILL.md +356 -0
- package/aris/skills/skills-codex/experiment-plan/SKILL.md +249 -0
- package/aris/skills/skills-codex/experiment-queue/SKILL.md +401 -0
- package/aris/skills/skills-codex/feishu-notify/SKILL.md +155 -0
- package/aris/skills/skills-codex/figure-description/SKILL.md +138 -0
- package/aris/skills/skills-codex/figure-spec/SKILL.md +252 -0
- package/aris/skills/skills-codex/formula-derivation/SKILL.md +280 -0
- package/aris/skills/skills-codex/gemini-search/SKILL.md +205 -0
- package/aris/skills/skills-codex/grant-proposal/SKILL.md +626 -0
- package/aris/skills/skills-codex/idea-creator/SKILL.md +438 -0
- package/aris/skills/skills-codex/idea-discovery/SKILL.md +477 -0
- package/aris/skills/skills-codex/idea-discovery-robot/SKILL.md +362 -0
- package/aris/skills/skills-codex/integrity-forensics/SKILL.md +108 -0
- package/aris/skills/skills-codex/interview-cheatsheet/SKILL.md +245 -0
- package/aris/skills/skills-codex/invention-structuring/SKILL.md +188 -0
- package/aris/skills/skills-codex/jurisdiction-format/SKILL.md +192 -0
- package/aris/skills/skills-codex/kill-argument/SKILL.md +403 -0
- package/aris/skills/skills-codex/mermaid-diagram/SKILL.md +379 -0
- package/aris/skills/skills-codex/meta-apply/SKILL.md +154 -0
- package/aris/skills/skills-codex/meta-optimize/SKILL.md +348 -0
- package/aris/skills/skills-codex/monitor-experiment/SKILL.md +98 -0
- package/aris/skills/skills-codex/novelty-check/SKILL.md +130 -0
- package/aris/skills/skills-codex/openalex/SKILL.md +228 -0
- package/aris/skills/skills-codex/overleaf-sync/SKILL.md +220 -0
- package/aris/skills/skills-codex/paper-claim-audit/SKILL.md +350 -0
- package/aris/skills/skills-codex/paper-compile/SKILL.md +253 -0
- package/aris/skills/skills-codex/paper-figure/SKILL.md +311 -0
- package/aris/skills/skills-codex/paper-illustration/SKILL.md +690 -0
- package/aris/skills/skills-codex/paper-illustration-image2/SKILL.md +383 -0
- package/aris/skills/skills-codex/paper-illustration-image2/scripts/paper_illustration_image2.py +255 -0
- package/aris/skills/skills-codex/paper-plan/SKILL.md +278 -0
- package/aris/skills/skills-codex/paper-poster/SKILL.md +19 -0
- package/aris/skills/skills-codex/paper-poster-html/SKILL.md +377 -0
- package/aris/skills/skills-codex/paper-slides/SKILL.md +571 -0
- package/aris/skills/skills-codex/paper-talk/SKILL.md +381 -0
- package/aris/skills/skills-codex/paper-write/SKILL.md +487 -0
- package/aris/skills/skills-codex/paper-write/templates/IEEEtran.bst +2409 -0
- package/aris/skills/skills-codex/paper-write/templates/IEEEtran.cls +6347 -0
- package/aris/skills/skills-codex/paper-write/templates/aaai2026.bst +1493 -0
- package/aris/skills/skills-codex/paper-write/templates/aaai2026.sty +315 -0
- package/aris/skills/skills-codex/paper-write/templates/aaai2026.tex +952 -0
- package/aris/skills/skills-codex/paper-write/templates/acl.sty +312 -0
- package/aris/skills/skills-codex/paper-write/templates/acl2026.tex +377 -0
- package/aris/skills/skills-codex/paper-write/templates/acl_natbib.bst +1940 -0
- package/aris/skills/skills-codex/paper-write/templates/acm.bst +3081 -0
- package/aris/skills/skills-codex/paper-write/templates/acm_mm2026.tex +204 -0
- package/aris/skills/skills-codex/paper-write/templates/acmart.cls +3520 -0
- package/aris/skills/skills-codex/paper-write/templates/cvpr.bst +1448 -0
- package/aris/skills/skills-codex/paper-write/templates/cvpr.sty +508 -0
- package/aris/skills/skills-codex/paper-write/templates/cvpr2026.tex +63 -0
- package/aris/skills/skills-codex/paper-write/templates/iclr2026.tex +84 -0
- package/aris/skills/skills-codex/paper-write/templates/iclr2026_conference.bst +1440 -0
- package/aris/skills/skills-codex/paper-write/templates/iclr2026_conference.sty +246 -0
- package/aris/skills/skills-codex/paper-write/templates/icml2026.sty +767 -0
- package/aris/skills/skills-codex/paper-write/templates/icml2026.tex +662 -0
- package/aris/skills/skills-codex/paper-write/templates/ieee_conference.tex +89 -0
- package/aris/skills/skills-codex/paper-write/templates/ieee_journal.tex +93 -0
- package/aris/skills/skills-codex/paper-write/templates/math_commands.tex +48 -0
- package/aris/skills/skills-codex/paper-write/templates/neurips2026.tex +493 -0
- package/aris/skills/skills-codex/paper-write/templates/neurips_2026.sty +437 -0
- package/aris/skills/skills-codex/paper-writing/SKILL.md +744 -0
- package/aris/skills/skills-codex/patent-novelty-check/SKILL.md +153 -0
- package/aris/skills/skills-codex/patent-pipeline/SKILL.md +344 -0
- package/aris/skills/skills-codex/patent-review/SKILL.md +202 -0
- package/aris/skills/skills-codex/pixel-art/SKILL.md +139 -0
- package/aris/skills/skills-codex/prior-art-search/SKILL.md +146 -0
- package/aris/skills/skills-codex/proof-checker/SKILL.md +554 -0
- package/aris/skills/skills-codex/proof-orchestrator/SKILL.md +260 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/audit-output-contract.md +126 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/deepseek-routing.md +76 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/dispatch-prompts.md +227 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/notation-audit.md +135 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/proof-audit-rubric.md +70 -0
- package/aris/skills/skills-codex/proof-orchestrator/references/stress-tests.md +38 -0
- package/aris/skills/skills-codex/proof-writer/SKILL.md +222 -0
- package/aris/skills/skills-codex/qzcli/SKILL.md +324 -0
- package/aris/skills/skills-codex/rebuttal/SKILL.md +305 -0
- package/aris/skills/skills-codex/render-html/SKILL.md +305 -0
- package/aris/skills/skills-codex/render-html/scripts/render_html.py +909 -0
- package/aris/skills/skills-codex/render-html/scripts/templates/academic.html +342 -0
- package/aris/skills/skills-codex/render-html/scripts/templates/dashboard.html +333 -0
- package/aris/skills/skills-codex/research-implement-feature/SKILL.md +406 -0
- package/aris/skills/skills-codex/research-lit/SKILL.md +464 -0
- package/aris/skills/skills-codex/research-pipeline/SKILL.md +346 -0
- package/aris/skills/skills-codex/research-refine/SKILL.md +721 -0
- package/aris/skills/skills-codex/research-refine-pipeline/SKILL.md +186 -0
- package/aris/skills/skills-codex/research-review/SKILL.md +160 -0
- package/aris/skills/skills-codex/research-wiki/SKILL.md +421 -0
- package/aris/skills/skills-codex/resubmit-pipeline/SKILL.md +444 -0
- package/aris/skills/skills-codex/result-to-claim/SKILL.md +246 -0
- package/aris/skills/skills-codex/run-experiment/SKILL.md +236 -0
- package/aris/skills/skills-codex/semantic-scholar/SKILL.md +219 -0
- package/aris/skills/skills-codex/serverless-modal/SKILL.md +335 -0
- package/aris/skills/skills-codex/shared-references/acceptance-gate.md +336 -0
- package/aris/skills/skills-codex/shared-references/assurance-contract.md +139 -0
- package/aris/skills/skills-codex/shared-references/capture-antipatterns.md +84 -0
- package/aris/skills/skills-codex/shared-references/citation-discipline.md +452 -0
- package/aris/skills/skills-codex/shared-references/compute-env-contract.md +163 -0
- package/aris/skills/skills-codex/shared-references/effort-contract.md +143 -0
- package/aris/skills/skills-codex/shared-references/evidence-precheck.md +73 -0
- package/aris/skills/skills-codex/shared-references/experiment-integrity.md +49 -0
- package/aris/skills/skills-codex/shared-references/external-cadence.md +334 -0
- package/aris/skills/skills-codex/shared-references/fan-out-pattern.md +375 -0
- package/aris/skills/skills-codex/shared-references/injection-hygiene.md +132 -0
- package/aris/skills/skills-codex/shared-references/integration-contract.md +372 -0
- package/aris/skills/skills-codex/shared-references/output-composition.md +98 -0
- package/aris/skills/skills-codex/shared-references/output-language.md +45 -0
- package/aris/skills/skills-codex/shared-references/output-manifest.md +40 -0
- package/aris/skills/skills-codex/shared-references/output-versioning.md +111 -0
- package/aris/skills/skills-codex/shared-references/patent-format-cn.md +199 -0
- package/aris/skills/skills-codex/shared-references/patent-format-ep.md +173 -0
- package/aris/skills/skills-codex/shared-references/patent-format-us.md +161 -0
- package/aris/skills/skills-codex/shared-references/patent-writing-principles.md +197 -0
- package/aris/skills/skills-codex/shared-references/prior-art-databases.md +141 -0
- package/aris/skills/skills-codex/shared-references/resumable-runs.md +125 -0
- package/aris/skills/skills-codex/shared-references/review-scope-limits.md +81 -0
- package/aris/skills/skills-codex/shared-references/review-tracing.md +144 -0
- package/aris/skills/skills-codex/shared-references/reviewer-independence.md +66 -0
- package/aris/skills/skills-codex/shared-references/reviewer-routing.md +128 -0
- package/aris/skills/skills-codex/shared-references/skill-governance.md +119 -0
- package/aris/skills/skills-codex/shared-references/taste-calibration.md +90 -0
- package/aris/skills/skills-codex/shared-references/venue-checklists.md +73 -0
- package/aris/skills/skills-codex/shared-references/wiki-helper-resolution.md +69 -0
- package/aris/skills/skills-codex/shared-references/writing-principles.md +525 -0
- package/aris/skills/skills-codex/slides-polish/SKILL.md +563 -0
- package/aris/skills/skills-codex/specification-writing/SKILL.md +211 -0
- package/aris/skills/skills-codex/system-profile/SKILL.md +103 -0
- package/aris/skills/skills-codex/training-check/SKILL.md +83 -0
- package/aris/skills/skills-codex/vast-gpu/SKILL.md +394 -0
- package/aris/skills/skills-codex/web-debug-search/SKILL.md +334 -0
- package/aris/skills/skills-codex/wiki-enrich/SKILL.md +255 -0
- package/aris/skills/skills-codex/writing-systems-papers/SKILL.md +184 -0
- package/aris/skills/skills-codex-claude-review/README.md +79 -0
- package/aris/skills/skills-codex-claude-review/README_CN.md +78 -0
- package/aris/skills/skills-codex-claude-review/auto-paper-improvement-loop/SKILL.md +625 -0
- package/aris/skills/skills-codex-claude-review/auto-review-loop/SKILL.md +516 -0
- package/{skills/novelty-check-claude → aris/skills/skills-codex-claude-review/novelty-check}/SKILL.md +2 -10
- package/aris/skills/skills-codex-claude-review/paper-figure/SKILL.md +319 -0
- package/aris/skills/skills-codex-claude-review/paper-plan/SKILL.md +287 -0
- package/aris/skills/skills-codex-claude-review/paper-write/SKILL.md +496 -0
- package/aris/skills/skills-codex-claude-review/research-refine/SKILL.md +732 -0
- package/aris/skills/skills-codex-claude-review/research-review/SKILL.md +177 -0
- package/aris/skills/skills-codex-gemini-review/README.md +176 -0
- package/aris/skills/skills-codex-gemini-review/README_CN.md +175 -0
- package/aris/skills/skills-codex-gemini-review/auto-paper-improvement-loop/SKILL.md +376 -0
- package/aris/skills/skills-codex-gemini-review/auto-review-loop/SKILL.md +304 -0
- package/aris/skills/skills-codex-gemini-review/grant-proposal/SKILL.md +630 -0
- package/aris/skills/skills-codex-gemini-review/idea-creator/SKILL.md +287 -0
- package/aris/skills/skills-codex-gemini-review/idea-discovery/SKILL.md +275 -0
- package/aris/skills/skills-codex-gemini-review/idea-discovery-robot/SKILL.md +365 -0
- package/aris/skills/skills-codex-gemini-review/novelty-check/SKILL.md +133 -0
- package/aris/skills/skills-codex-gemini-review/paper-figure/SKILL.md +289 -0
- package/aris/skills/skills-codex-gemini-review/paper-plan/SKILL.md +265 -0
- package/aris/skills/skills-codex-gemini-review/paper-poster-html/SKILL.md +102 -0
- package/aris/skills/skills-codex-gemini-review/paper-slides/SKILL.md +582 -0
- package/aris/skills/skills-codex-gemini-review/paper-write/SKILL.md +422 -0
- package/aris/skills/skills-codex-gemini-review/paper-writing/SKILL.md +325 -0
- package/aris/skills/skills-codex-gemini-review/research-refine/SKILL.md +674 -0
- package/aris/skills/skills-codex-gemini-review/research-review/SKILL.md +137 -0
- package/aris/skills/slides-polish/SKILL.md +565 -0
- package/aris/skills/specification-writing/SKILL.md +211 -0
- package/aris/skills/system-profile/SKILL.md +103 -0
- package/aris/skills/training-check/SKILL.md +132 -0
- package/aris/skills/vast-gpu/SKILL.md +394 -0
- package/aris/skills/web-debug-search/SKILL.md +334 -0
- package/aris/skills/wiki-enrich/SKILL.md +257 -0
- package/aris/skills/writing-systems-papers/SKILL.md +184 -0
- package/aris/templates/CLAUDE_MD_TEMPLATE.md +29 -0
- package/aris/templates/EXPERIMENT_LOG_TEMPLATE.md +47 -0
- package/aris/templates/EXPERIMENT_PLAN_TEMPLATE.md +51 -0
- package/aris/templates/EXPERIMENT_PLAN_TEMPLATE_CN.md +53 -0
- package/aris/templates/FINDINGS_TEMPLATE.md +52 -0
- package/aris/templates/IDEA_CANDIDATES_TEMPLATE.md +47 -0
- package/aris/templates/IDEA_CANDIDATES_TEMPLATE_CN.md +47 -0
- package/aris/templates/INVENTION_BRIEF_TEMPLATE.md +87 -0
- package/aris/templates/MANIFEST_TEMPLATE.md +7 -0
- package/aris/templates/NARRATIVE_REPORT_TEMPLATE.md +49 -0
- package/aris/templates/PAPER_PLAN_TEMPLATE.md +47 -0
- package/aris/templates/PATENT_CLAIMS_TEMPLATE.md +78 -0
- package/aris/templates/PATENT_SPECIFICATION_TEMPLATE.md +68 -0
- package/aris/templates/README.md +57 -0
- package/aris/templates/RESEARCH_BRIEF_TEMPLATE.md +35 -0
- package/aris/templates/RESEARCH_BRIEF_TEMPLATE_CN.md +41 -0
- package/aris/templates/RESEARCH_CONTRACT_TEMPLATE.md +60 -0
- package/aris/templates/claude-hooks/corpus_write_guard.json +16 -0
- package/aris/templates/claude-hooks/corpus_write_guard.py +85 -0
- package/aris/templates/claude-hooks/meta_logging.json +74 -0
- package/aris/templates/gitignore-trace.txt +3 -0
- package/aris/tests/__init__.py +0 -0
- package/aris/tests/_feishu_bridge_helpers.py +82 -0
- package/aris/tests/_llm_chat_helpers.py +179 -0
- package/aris/tests/_minimax_helpers.py +189 -0
- package/aris/tests/test_argument_hint_lint.py +183 -0
- package/aris/tests/test_arxiv_fetch.py +266 -0
- package/aris/tests/test_auto_proceed_contract.py +106 -0
- package/aris/tests/test_capture_filter.py +86 -0
- package/aris/tests/test_claude_review_artifact_tools.py +171 -0
- package/aris/tests/test_claude_review_generator.py +51 -0
- package/aris/tests/test_claude_review_overlay.py +64 -0
- package/aris/tests/test_claude_review_parse.py +290 -0
- package/aris/tests/test_codex_exec_bridge.py +306 -0
- package/aris/tests/test_codex_image2_server.py +424 -0
- package/aris/tests/test_codex_install_update.py +511 -0
- package/aris/tests/test_codex_skill_mirror.py +547 -0
- package/aris/tests/test_copilot_install.py +1202 -0
- package/aris/tests/test_copilot_native_evidence.py +562 -0
- package/aris/tests/test_corpus_write_guard.py +76 -0
- package/aris/tests/test_deepxiv_fetch.py +180 -0
- package/aris/tests/test_evidence_check.py +189 -0
- package/aris/tests/test_exa_search.py +220 -0
- package/aris/tests/test_feishu_bridge_server.py +289 -0
- package/aris/tests/test_forensics_gate.py +565 -0
- package/aris/tests/test_http_reviewer_fallback_contract.py +43 -0
- package/aris/tests/test_http_reviewer_provenance_alignment.py +25 -0
- package/aris/tests/test_http_reviewer_review_gate.py +112 -0
- package/aris/tests/test_idea_creator_query_pack_scan.py +238 -0
- package/aris/tests/test_idea_discovery_gate.py +400 -0
- package/aris/tests/test_install_aris_agent_profiles.py +98 -0
- package/aris/tests/test_install_aris_ps1.py +359 -0
- package/aris/tests/test_install_aris_replace_link.py +119 -0
- package/aris/tests/test_install_aris_selective.py +235 -0
- package/aris/tests/test_install_aris_tools_symlink.py +196 -0
- package/aris/tests/test_iteration_log.py +85 -0
- package/aris/tests/test_llm_chat_converter.py +61 -0
- package/aris/tests/test_llm_chat_review_fallback.py +516 -0
- package/aris/tests/test_llm_chat_server.py +414 -0
- package/aris/tests/test_mainline_large_payload_prompts.py +72 -0
- package/aris/tests/test_manual_review.py +804 -0
- package/aris/tests/test_manual_review_integration.py +305 -0
- package/aris/tests/test_minimax_chat_server.py +464 -0
- package/aris/tests/test_minimax_integration.py +241 -0
- package/aris/tests/test_openalex_fetch.py +197 -0
- package/aris/tests/test_plugin_manifest.py +40 -0
- package/aris/tests/test_proof_suite_integration.py +66 -0
- package/aris/tests/test_provenance.py +284 -0
- package/aris/tests/test_research_wiki_claims.py +93 -0
- package/aris/tests/test_research_wiki_encoding.py +74 -0
- package/aris/tests/test_research_wiki_experiments.py +88 -0
- package/aris/tests/test_research_wiki_fetch_arxiv_metadata.py +242 -0
- package/aris/tests/test_research_wiki_helper_resolution.py +253 -0
- package/aris/tests/test_research_wiki_ideas.py +113 -0
- package/aris/tests/test_review_gate.py +358 -0
- package/aris/tests/test_reviewer_pins.py +90 -0
- package/aris/tests/test_run_state.py +313 -0
- package/aris/tests/test_skill_groups.py +111 -0
- package/aris/tests/test_skill_picker.py +207 -0
- package/aris/tests/test_threat_scan.py +139 -0
- package/aris/tests/test_trigger_eval.py +171 -0
- package/aris/tests/test_verify_paper_audits.py +198 -0
- package/aris/tests/test_watchdog.py +496 -0
- package/aris/tests/test_watchdog_loop.py +148 -0
- package/aris/tests/test_web_debug_search.py +157 -0
- package/aris/tools/arxiv_fetch.py +311 -0
- package/aris/tools/capture_filter.py +126 -0
- package/aris/tools/check_skills_inventory.py +287 -0
- package/aris/tools/convert_skills_to_llm_chat.py +282 -0
- package/aris/tools/copilot_native_evidence.py +818 -0
- package/aris/tools/deepxiv_fetch.py +213 -0
- package/aris/tools/evidence_check.py +212 -0
- package/aris/tools/exa_search.py +425 -0
- package/aris/tools/experiment_queue/README.md +118 -0
- package/aris/tools/experiment_queue/build_manifest.py +44 -0
- package/aris/tools/experiment_queue/queue_manager.py +44 -0
- package/aris/tools/extract_paper_style.py +560 -0
- package/aris/tools/figure_renderer.py +69 -0
- package/aris/tools/forensics_gate.py +669 -0
- package/aris/tools/generate_codex_claude_review_overrides.py +355 -0
- package/aris/tools/idea_discovery_gate.py +256 -0
- package/aris/tools/install_aris.ps1 +1372 -0
- package/aris/tools/install_aris.sh +1416 -0
- package/aris/tools/install_aris_codex.sh +1023 -0
- package/aris/tools/install_aris_copilot.sh +1052 -0
- package/aris/tools/iteration_log.py +143 -0
- package/aris/tools/lint_skills_helpers.sh +84 -0
- package/aris/tools/meta_opt/check_ready.sh +80 -0
- package/aris/tools/meta_opt/log_event.sh +91 -0
- package/aris/tools/meta_opt/trigger_eval.py +280 -0
- package/aris/tools/meta_opt/trigger_evals.sample.json +28 -0
- package/aris/tools/openalex_fetch.py +326 -0
- package/aris/tools/overleaf_audit.sh +104 -0
- package/aris/tools/overleaf_setup.sh +150 -0
- package/aris/tools/paper_illustration_image2.py +62 -0
- package/aris/tools/provenance.py +294 -0
- package/aris/tools/research_wiki.py +1720 -0
- package/aris/tools/review_gate.py +502 -0
- package/aris/tools/run_state.py +399 -0
- package/aris/tools/save_trace.sh +477 -0
- package/aris/tools/semantic_scholar_fetch.py +438 -0
- package/aris/tools/skill-groups.tsv +117 -0
- package/aris/tools/skill_picker.py +238 -0
- package/aris/tools/smart_update.ps1 +521 -0
- package/aris/tools/smart_update.sh +591 -0
- package/aris/tools/smart_update_codex.sh +419 -0
- package/aris/tools/smart_update_copilot.sh +605 -0
- package/aris/tools/threat_scan.py +222 -0
- package/aris/tools/verify_paper_audits.sh +487 -0
- package/aris/tools/verify_papers.py +613 -0
- package/aris/tools/verify_wiki_coverage.sh +176 -0
- package/aris/tools/watchdog.py +485 -0
- package/aris/xhs_post.md +125 -0
- package/dsh/client.js +43 -14
- package/dsh/http-rpc.mjs +8 -7
- package/mcp-servers/codex-exec/README.md +4 -4
- package/mcp-servers/codex-exec/server.py +1 -1
- package/package.json +4 -5
- package/skills/ablation-planner/SKILL.md +1 -1
- package/skills/auto-paper-improvement-loop/SKILL.md +10 -10
- package/skills/auto-paper-improvement-loop-claude/SKILL.md +25 -612
- package/skills/auto-review-loop/SKILL.md +4 -4
- package/skills/auto-review-loop-claude/SKILL.md +25 -503
- package/skills/citation-audit/SKILL.md +3 -3
- package/skills/claims-drafting/SKILL.md +2 -2
- package/skills/experiment-audit/SKILL.md +4 -4
- package/skills/experiment-bridge/SKILL.md +14 -12
- package/skills/figure-spec/SKILL.md +1 -1
- package/skills/grant-proposal/SKILL.md +7 -7
- package/skills/human-in-loop-refine/SKILL.md +119 -0
- package/skills/human-in-loop-refine/scripts/refine_round.py +449 -0
- package/skills/human-in-loop-refine/scripts/test_refine_round.py +200 -0
- package/skills/human-in-loop-refine/templates/INPUTS.md +20 -0
- package/skills/human-in-loop-refine/templates/PAPER_NOTE.md +29 -0
- package/skills/human-in-loop-refine/templates/REVISION.md +27 -0
- package/skills/idea-creator/SKILL.md +2 -2
- package/skills/idea-discovery/SKILL.md +15 -13
- package/skills/idea-discovery-robot/SKILL.md +11 -9
- package/skills/integrity-forensics/SKILL.md +1 -1
- package/skills/interview-cheatsheet/SKILL.md +3 -3
- package/skills/invention-structuring/SKILL.md +2 -2
- package/skills/kill-argument/SKILL.md +8 -8
- package/skills/kill-argument-claude/SKILL.md +30 -431
- package/skills/meta-apply/SKILL.md +1 -1
- package/skills/meta-optimize/SKILL.md +2 -2
- package/skills/novelty-check/SKILL.md +2 -2
- package/skills/paper-claim-audit/SKILL.md +5 -5
- package/skills/paper-figure/SKILL.md +4 -3
- package/skills/paper-figure-claude/SKILL.md +24 -305
- package/skills/paper-plan/SKILL.md +3 -3
- package/skills/paper-plan-claude/SKILL.md +24 -273
- package/skills/paper-poster-html/SKILL.md +11 -9
- package/skills/paper-slides/SKILL.md +5 -5
- package/skills/paper-talk/SKILL.md +10 -8
- package/skills/paper-write/SKILL.md +3 -3
- package/skills/paper-write-claude/SKILL.md +24 -482
- package/skills/paper-writing/SKILL.md +19 -17
- package/skills/patent-novelty-check/SKILL.md +3 -3
- package/skills/patent-pipeline/SKILL.md +2 -2
- package/skills/patent-review/SKILL.md +3 -3
- package/skills/proof-checker/SKILL.md +12 -12
- package/skills/proof-checker-claude/SKILL.md +30 -860
- package/skills/rebuttal/SKILL.md +12 -10
- package/skills/render-html/SKILL.md +1 -1
- package/skills/research-implement-feature/SKILL.md +1 -1
- package/skills/research-pipeline/SKILL.md +15 -13
- package/skills/research-refine/SKILL.md +6 -6
- package/skills/research-refine-claude/SKILL.md +24 -719
- package/skills/research-refine-pipeline/SKILL.md +10 -8
- package/skills/research-review/SKILL.md +7 -7
- package/skills/research-review-claude/SKILL.md +25 -164
- package/skills/resubmit-pipeline/SKILL.md +14 -12
- package/skills/result-to-claim/SKILL.md +2 -2
- package/skills/shared-references/assurance-contract.md +1 -1
- package/skills/shared-references/effort-contract.md +1 -1
- package/skills/shared-references/fan-out-pattern.md +4 -4
- package/skills/shared-references/resumable-runs.md +2 -2
- package/skills/shared-references/review-scope-limits.md +1 -1
- package/skills/shared-references/review-tracing.md +2 -2
- package/skills/shared-references/reviewer-routing.md +16 -14
- package/skills/shared-references/skill-governance.md +3 -3
- package/skills/skills-codex/ablation-planner/SKILL.md +1 -1
- package/skills/skills-codex/auto-paper-improvement-loop/SKILL.md +8 -8
- package/skills/skills-codex/auto-review-loop/SKILL.md +2 -2
- package/skills/skills-codex/citation-audit/SKILL.md +4 -4
- package/skills/skills-codex/claims-drafting/SKILL.md +2 -2
- package/skills/skills-codex/experiment-audit/SKILL.md +9 -9
- package/skills/skills-codex/experiment-bridge/SKILL.md +1 -1
- package/skills/skills-codex/figure-spec/SKILL.md +1 -1
- package/skills/skills-codex/grant-proposal/SKILL.md +7 -7
- package/skills/skills-codex/idea-creator/SKILL.md +4 -4
- package/skills/skills-codex/idea-discovery/SKILL.md +6 -6
- package/skills/skills-codex/idea-discovery-robot/SKILL.md +1 -1
- package/skills/skills-codex/integrity-forensics/SKILL.md +4 -2
- package/skills/skills-codex/interview-cheatsheet/SKILL.md +3 -3
- package/skills/skills-codex/invention-structuring/SKILL.md +2 -2
- package/skills/skills-codex/kill-argument/SKILL.md +7 -7
- package/skills/skills-codex/meta-apply/SKILL.md +1 -1
- package/skills/skills-codex/meta-optimize/SKILL.md +2 -2
- package/skills/skills-codex/novelty-check/SKILL.md +1 -1
- package/skills/skills-codex/paper-claim-audit/SKILL.md +6 -6
- package/skills/skills-codex/paper-figure/SKILL.md +3 -3
- package/skills/skills-codex/paper-plan/SKILL.md +3 -3
- package/skills/skills-codex/paper-poster-html/SKILL.md +3 -3
- package/skills/skills-codex/paper-slides/SKILL.md +4 -4
- package/skills/skills-codex/paper-write/SKILL.md +3 -3
- package/skills/skills-codex/paper-writing/SKILL.md +7 -7
- package/skills/skills-codex/patent-novelty-check/SKILL.md +3 -3
- package/skills/skills-codex/patent-pipeline/SKILL.md +2 -2
- package/skills/skills-codex/patent-review/SKILL.md +3 -3
- package/skills/skills-codex/proof-checker/SKILL.md +11 -11
- package/skills/skills-codex/rebuttal/SKILL.md +2 -2
- package/skills/skills-codex/render-html/SKILL.md +1 -1
- package/skills/skills-codex/research-implement-feature/SKILL.md +1 -1
- package/skills/skills-codex/research-pipeline/SKILL.md +5 -5
- package/skills/skills-codex/research-refine/SKILL.md +6 -6
- package/skills/skills-codex/research-review/SKILL.md +6 -6
- package/skills/skills-codex/resubmit-pipeline/SKILL.md +4 -4
- package/skills/skills-codex/result-to-claim/SKILL.md +2 -2
- package/skills/skills-codex/shared-references/assurance-contract.md +2 -2
- package/skills/skills-codex/shared-references/resumable-runs.md +3 -3
- package/skills/skills-codex/shared-references/review-scope-limits.md +1 -1
- package/skills/skills-codex/shared-references/review-tracing.md +3 -3
- package/skills/skills-codex/shared-references/reviewer-independence.md +2 -2
- package/skills/skills-codex/shared-references/reviewer-routing.md +10 -10
- package/skills/skills-codex/slides-polish/SKILL.md +3 -3
- package/skills/skills-codex/specification-writing/SKILL.md +2 -2
- package/skills/slides-polish/SKILL.md +3 -3
- package/skills/specification-writing/SKILL.md +2 -2
- package/skills/training-check/SKILL.md +2 -2
- package/skills/verify-manager/SKILL.md +88 -0
- package/skills/verify-manager/phases/1-specify.md +17 -0
- package/skills/verify-manager/phases/2-plan.md +14 -0
- package/skills/verify-manager/phases/3-tasks.md +19 -0
- package/skills/verify-manager/phases/4-run.md +12 -0
- package/skills/verify-manager/phases/5-analyze.md +21 -0
- package/skills/verify-manager/phases/6-next.md +19 -0
- package/skills/verify-manager/scripts/test_verify_space.py +202 -0
- package/skills/verify-manager/scripts/verify_space.py +451 -0
- package/skills/verify-manager/templates/HYPOTHESIS.md +37 -0
- package/skills/verify-manager/templates/NEXT_IDEAS.md +20 -0
- package/skills/verify-manager/templates/PILOT_PLAN.md +56 -0
- package/skills/verify-manager/templates/PILOT_TASKS.md +28 -0
- package/skills/verify-manager/templates/PROMOTION.md +17 -0
- package/skills/verify-manager/templates/RESULT.md +36 -0
- package/test/client-render-smoke.mjs +143 -0
- package/test/test_claude_overlays.py +99 -0
- package/test/transport-smoke.mjs +16 -4
- package/tools/__pycache__/build_claude_overlays.cpython-314.pyc +0 -0
- package/tools/__pycache__/verify_lane.cpython-314.pyc +0 -0
- package/tools/build_claude_overlays.py +187 -0
- package/tools/check_skills_inventory.py +17 -3
- package/tools/convert_skills_to_llm_chat.py +15 -15
- package/tools/generate_codex_claude_review_overrides.py +12 -12
- package/tools/run_state.py +1 -1
- package/tools/save_trace.sh +2 -2
|
@@ -0,0 +1,755 @@
|
|
|
1
|
+
<!DOCTYPE html>
|
|
2
|
+
<html lang="zh-CN">
|
|
3
|
+
<head>
|
|
4
|
+
<meta charset="UTF-8">
|
|
5
|
+
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
6
|
+
<title>Flow Matching Quick Reference</title>
|
|
7
|
+
|
|
8
|
+
<meta name="generator" content="ARIS render-html (academic, v1)">
|
|
9
|
+
<meta name="aris:source-path" content="docs/tutorials/flow_matching_tutorial.md">
|
|
10
|
+
<meta name="aris:source-sha256" content="ab4358901d204fec676df6ce3b8559be4e69fec3833f764bcfc937b113eea281">
|
|
11
|
+
<meta name="aris:generated-at" content="2026-05-19 03:40 UTC">
|
|
12
|
+
|
|
13
|
+
<!-- MathJax 3 -->
|
|
14
|
+
<script>
|
|
15
|
+
window.MathJax = {
|
|
16
|
+
tex: { inlineMath: [['$', '$'], ['\\(', '\\)']], displayMath: [['$$', '$$'], ['\\[', '\\]']], processEscapes: true },
|
|
17
|
+
options: { skipHtmlTags: ['script', 'noscript', 'style', 'textarea', 'pre', 'code'] }
|
|
18
|
+
};
|
|
19
|
+
</script>
|
|
20
|
+
<script src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js" async></script>
|
|
21
|
+
|
|
22
|
+
<!-- highlight.js -->
|
|
23
|
+
<link rel="stylesheet" href="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.9.0/build/styles/atom-one-light.min.css">
|
|
24
|
+
<script src="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.9.0/build/highlight.min.js"></script>
|
|
25
|
+
<script>document.addEventListener('DOMContentLoaded', () => hljs.highlightAll());</script>
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
<style>
|
|
29
|
+
:root {
|
|
30
|
+
--bg: #fdfcf7;
|
|
31
|
+
--bg-soft: #f4f1ea;
|
|
32
|
+
--bg-code: #f8f5ec;
|
|
33
|
+
--ink: #1a1a1a;
|
|
34
|
+
--ink-soft: #4a4a4a;
|
|
35
|
+
--ink-muted: #6b6b6b;
|
|
36
|
+
--primary: #1a4a8c;
|
|
37
|
+
--primary-soft: #2d6cb8;
|
|
38
|
+
--accent: #b8390e;
|
|
39
|
+
--warn: #b45309;
|
|
40
|
+
--warn-bg: #fef3c7;
|
|
41
|
+
--info-bg: #dbeafe;
|
|
42
|
+
--good-bg: #d1fae5;
|
|
43
|
+
--good: #065f46;
|
|
44
|
+
--bad-bg: #fee2e2;
|
|
45
|
+
--bad: #991b1b;
|
|
46
|
+
--border: #d6d0c0;
|
|
47
|
+
--border-soft: #e8e3d5;
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
* { box-sizing: border-box; }
|
|
51
|
+
html { scroll-behavior: smooth; }
|
|
52
|
+
|
|
53
|
+
body {
|
|
54
|
+
font-family: "Source Serif Pro", "Source Serif 4", "Crimson Pro", "Georgia", "Songti SC", "STSong", serif;
|
|
55
|
+
line-height: 1.65;
|
|
56
|
+
color: var(--ink);
|
|
57
|
+
background: var(--bg);
|
|
58
|
+
margin: 0;
|
|
59
|
+
padding: 0;
|
|
60
|
+
font-size: 16px;
|
|
61
|
+
}
|
|
62
|
+
|
|
63
|
+
.layout {
|
|
64
|
+
max-width: 1280px;
|
|
65
|
+
margin: 0 auto;
|
|
66
|
+
display: grid;
|
|
67
|
+
grid-template-columns: 260px 1fr;
|
|
68
|
+
gap: 48px;
|
|
69
|
+
padding: 40px 32px;
|
|
70
|
+
}
|
|
71
|
+
|
|
72
|
+
nav.toc {
|
|
73
|
+
position: sticky;
|
|
74
|
+
top: 24px;
|
|
75
|
+
align-self: start;
|
|
76
|
+
font-size: 13px;
|
|
77
|
+
max-height: calc(100vh - 48px);
|
|
78
|
+
overflow-y: auto;
|
|
79
|
+
border-right: 1px solid var(--border-soft);
|
|
80
|
+
padding-right: 16px;
|
|
81
|
+
}
|
|
82
|
+
nav.toc h3 {
|
|
83
|
+
margin: 0 0 12px;
|
|
84
|
+
font-size: 12px;
|
|
85
|
+
text-transform: uppercase;
|
|
86
|
+
letter-spacing: 0.08em;
|
|
87
|
+
color: var(--ink-muted);
|
|
88
|
+
font-weight: 600;
|
|
89
|
+
}
|
|
90
|
+
nav.toc ol { list-style: none; padding: 0; margin: 0; counter-reset: toc; }
|
|
91
|
+
nav.toc ol li { margin: 5px 0; counter-increment: toc; }
|
|
92
|
+
nav.toc ol li::before { content: counter(toc) ". "; color: var(--ink-muted); margin-right: 4px; }
|
|
93
|
+
nav.toc a {
|
|
94
|
+
color: var(--ink-soft);
|
|
95
|
+
text-decoration: none;
|
|
96
|
+
border-bottom: 1px dotted transparent;
|
|
97
|
+
}
|
|
98
|
+
nav.toc a:hover { color: var(--primary); border-bottom-color: var(--primary); }
|
|
99
|
+
nav.toc ul { list-style: none; padding-left: 14px; margin: 3px 0; font-size: 12px; }
|
|
100
|
+
nav.toc ul li::before { content: "→ "; color: var(--border); }
|
|
101
|
+
|
|
102
|
+
main { min-width: 0; }
|
|
103
|
+
|
|
104
|
+
header.hero {
|
|
105
|
+
border-bottom: 3px double var(--primary);
|
|
106
|
+
padding-bottom: 24px;
|
|
107
|
+
margin-bottom: 32px;
|
|
108
|
+
}
|
|
109
|
+
header.hero .eyebrow {
|
|
110
|
+
color: var(--accent);
|
|
111
|
+
font-size: 13px;
|
|
112
|
+
text-transform: uppercase;
|
|
113
|
+
letter-spacing: 0.12em;
|
|
114
|
+
font-weight: 600;
|
|
115
|
+
margin-bottom: 8px;
|
|
116
|
+
}
|
|
117
|
+
header.hero h1 {
|
|
118
|
+
font-size: 32px;
|
|
119
|
+
line-height: 1.2;
|
|
120
|
+
margin: 0 0 12px;
|
|
121
|
+
color: var(--ink);
|
|
122
|
+
font-weight: 700;
|
|
123
|
+
letter-spacing: -0.01em;
|
|
124
|
+
}
|
|
125
|
+
header.hero .subtitle {
|
|
126
|
+
font-size: 16px;
|
|
127
|
+
color: var(--ink-soft);
|
|
128
|
+
margin: 0 0 8px;
|
|
129
|
+
font-style: italic;
|
|
130
|
+
}
|
|
131
|
+
header.hero .byline {
|
|
132
|
+
font-size: 14px;
|
|
133
|
+
color: var(--ink-soft);
|
|
134
|
+
margin: 0 0 20px;
|
|
135
|
+
}
|
|
136
|
+
header.hero .byline strong {
|
|
137
|
+
color: var(--ink);
|
|
138
|
+
font-weight: 600;
|
|
139
|
+
}
|
|
140
|
+
header.hero .meta {
|
|
141
|
+
display: flex;
|
|
142
|
+
gap: 20px;
|
|
143
|
+
flex-wrap: wrap;
|
|
144
|
+
font-size: 12px;
|
|
145
|
+
color: var(--ink-muted);
|
|
146
|
+
border-top: 1px solid var(--border-soft);
|
|
147
|
+
padding-top: 14px;
|
|
148
|
+
}
|
|
149
|
+
header.hero .meta span strong { color: var(--ink-soft); }
|
|
150
|
+
header.hero .meta code {
|
|
151
|
+
font-family: "JetBrains Mono", "SF Mono", "Menlo", "Consolas", monospace;
|
|
152
|
+
font-size: 11px;
|
|
153
|
+
background: var(--bg-soft);
|
|
154
|
+
padding: 1px 5px;
|
|
155
|
+
border-radius: 3px;
|
|
156
|
+
border: 1px solid var(--border-soft);
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
h2 {
|
|
160
|
+
font-size: 24px;
|
|
161
|
+
margin: 44px 0 14px;
|
|
162
|
+
padding-bottom: 8px;
|
|
163
|
+
border-bottom: 1px solid var(--border);
|
|
164
|
+
color: var(--ink);
|
|
165
|
+
font-weight: 700;
|
|
166
|
+
}
|
|
167
|
+
h2 .num { color: var(--primary); font-weight: 600; margin-right: 8px; }
|
|
168
|
+
h3 { font-size: 19px; margin: 28px 0 10px; color: var(--primary); font-weight: 600; }
|
|
169
|
+
h4 { font-size: 16px; margin: 20px 0 8px; color: var(--ink); font-weight: 600; }
|
|
170
|
+
|
|
171
|
+
p { margin: 10px 0; }
|
|
172
|
+
ul, ol { padding-left: 22px; margin: 10px 0; }
|
|
173
|
+
ul li, ol li { margin: 4px 0; }
|
|
174
|
+
ul li::marker { color: var(--primary); }
|
|
175
|
+
|
|
176
|
+
strong { color: var(--accent); font-weight: 600; }
|
|
177
|
+
em { color: var(--ink-soft); }
|
|
178
|
+
|
|
179
|
+
a { color: var(--primary); }
|
|
180
|
+
a:hover { color: var(--accent); }
|
|
181
|
+
|
|
182
|
+
code:not(.hljs) {
|
|
183
|
+
font-family: "JetBrains Mono", "SF Mono", "Menlo", "Consolas", monospace;
|
|
184
|
+
font-size: 0.86em;
|
|
185
|
+
background: var(--bg-code);
|
|
186
|
+
padding: 1px 5px;
|
|
187
|
+
border-radius: 3px;
|
|
188
|
+
border: 1px solid var(--border-soft);
|
|
189
|
+
color: var(--accent);
|
|
190
|
+
}
|
|
191
|
+
|
|
192
|
+
pre {
|
|
193
|
+
background: #fafaf6;
|
|
194
|
+
border: 1px solid var(--border);
|
|
195
|
+
border-left: 4px solid var(--primary);
|
|
196
|
+
padding: 0;
|
|
197
|
+
overflow-x: auto;
|
|
198
|
+
border-radius: 4px;
|
|
199
|
+
margin: 14px 0;
|
|
200
|
+
}
|
|
201
|
+
pre code, pre code.hljs {
|
|
202
|
+
background: transparent !important;
|
|
203
|
+
display: block;
|
|
204
|
+
padding: 14px 18px !important;
|
|
205
|
+
font-size: 13px;
|
|
206
|
+
line-height: 1.55;
|
|
207
|
+
font-family: "JetBrains Mono", "SF Mono", "Menlo", monospace;
|
|
208
|
+
color: var(--ink);
|
|
209
|
+
}
|
|
210
|
+
pre.diagram {
|
|
211
|
+
background: #f9f6ed;
|
|
212
|
+
border-left: 4px solid var(--accent);
|
|
213
|
+
font-size: 12.5px;
|
|
214
|
+
line-height: 1.4;
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
.callout {
|
|
218
|
+
margin: 16px 0;
|
|
219
|
+
padding: 12px 16px;
|
|
220
|
+
border-radius: 4px;
|
|
221
|
+
border-left: 4px solid;
|
|
222
|
+
font-size: 15px;
|
|
223
|
+
}
|
|
224
|
+
.callout-title {
|
|
225
|
+
font-weight: 600;
|
|
226
|
+
margin-bottom: 6px;
|
|
227
|
+
font-size: 12px;
|
|
228
|
+
text-transform: uppercase;
|
|
229
|
+
letter-spacing: 0.06em;
|
|
230
|
+
}
|
|
231
|
+
.callout-info { background: var(--info-bg); border-left-color: var(--primary); }
|
|
232
|
+
.callout-info .callout-title { color: var(--primary); }
|
|
233
|
+
.callout-warn { background: var(--warn-bg); border-left-color: var(--warn); }
|
|
234
|
+
.callout-warn .callout-title { color: var(--warn); }
|
|
235
|
+
.callout-good { background: var(--good-bg); border-left-color: var(--good); }
|
|
236
|
+
.callout-good .callout-title { color: var(--good); }
|
|
237
|
+
.callout-bad { background: var(--bad-bg); border-left-color: var(--bad); }
|
|
238
|
+
.callout-bad .callout-title { color: var(--bad); }
|
|
239
|
+
|
|
240
|
+
table {
|
|
241
|
+
width: 100%;
|
|
242
|
+
border-collapse: collapse;
|
|
243
|
+
margin: 16px 0;
|
|
244
|
+
font-size: 14px;
|
|
245
|
+
border: 1px solid var(--border);
|
|
246
|
+
border-radius: 4px;
|
|
247
|
+
overflow: hidden;
|
|
248
|
+
}
|
|
249
|
+
thead { background: var(--primary); color: white; }
|
|
250
|
+
th, td {
|
|
251
|
+
text-align: left;
|
|
252
|
+
padding: 9px 12px;
|
|
253
|
+
border-bottom: 1px solid var(--border-soft);
|
|
254
|
+
vertical-align: top;
|
|
255
|
+
}
|
|
256
|
+
th { font-weight: 600; font-size: 13px; letter-spacing: 0.02em; }
|
|
257
|
+
tr:last-child td { border-bottom: none; }
|
|
258
|
+
tbody tr:nth-child(even) { background: var(--bg-soft); }
|
|
259
|
+
|
|
260
|
+
details.qa, details {
|
|
261
|
+
background: white;
|
|
262
|
+
border: 1px solid var(--border-soft);
|
|
263
|
+
border-radius: 6px;
|
|
264
|
+
margin: 10px 0;
|
|
265
|
+
padding: 0;
|
|
266
|
+
}
|
|
267
|
+
details summary {
|
|
268
|
+
cursor: pointer;
|
|
269
|
+
padding: 10px 14px;
|
|
270
|
+
font-weight: 600;
|
|
271
|
+
font-size: 14px;
|
|
272
|
+
color: var(--primary);
|
|
273
|
+
list-style: none;
|
|
274
|
+
user-select: none;
|
|
275
|
+
}
|
|
276
|
+
details summary::-webkit-details-marker { display: none; }
|
|
277
|
+
details summary::before {
|
|
278
|
+
content: "▸ ";
|
|
279
|
+
margin-right: 4px;
|
|
280
|
+
display: inline-block;
|
|
281
|
+
transition: transform 0.15s;
|
|
282
|
+
}
|
|
283
|
+
details[open] summary::before { transform: rotate(90deg); }
|
|
284
|
+
details[open] summary { border-bottom: 1px solid var(--border-soft); }
|
|
285
|
+
details > :not(summary) { padding: 10px 14px; }
|
|
286
|
+
details p:first-of-type { margin-top: 8px; }
|
|
287
|
+
|
|
288
|
+
mjx-container[display="true"] { margin: 12px 0 !important; }
|
|
289
|
+
|
|
290
|
+
footer.aris-footer {
|
|
291
|
+
margin-top: 60px;
|
|
292
|
+
padding-top: 20px;
|
|
293
|
+
border-top: 1px solid var(--border);
|
|
294
|
+
font-size: 12px;
|
|
295
|
+
color: var(--ink-muted);
|
|
296
|
+
}
|
|
297
|
+
footer.aris-footer a { color: var(--ink-muted); border-bottom: 1px dotted var(--border); }
|
|
298
|
+
|
|
299
|
+
@media (max-width: 900px) {
|
|
300
|
+
.layout { grid-template-columns: 1fr; gap: 20px; padding: 20px 16px; }
|
|
301
|
+
nav.toc {
|
|
302
|
+
position: static;
|
|
303
|
+
max-height: none;
|
|
304
|
+
border-right: none;
|
|
305
|
+
border-bottom: 1px solid var(--border-soft);
|
|
306
|
+
padding-right: 0;
|
|
307
|
+
padding-bottom: 14px;
|
|
308
|
+
}
|
|
309
|
+
header.hero h1 { font-size: 24px; }
|
|
310
|
+
h2 { font-size: 20px; }
|
|
311
|
+
}
|
|
312
|
+
@media print {
|
|
313
|
+
nav.toc { display: none; }
|
|
314
|
+
.layout { grid-template-columns: 1fr; padding: 0; }
|
|
315
|
+
body { background: white; }
|
|
316
|
+
header.hero { border-bottom-color: var(--ink); }
|
|
317
|
+
}
|
|
318
|
+
</style>
|
|
319
|
+
</head>
|
|
320
|
+
<body>
|
|
321
|
+
<div class="layout">
|
|
322
|
+
|
|
323
|
+
<nav class="toc">
|
|
324
|
+
<h3>Contents</h3>
|
|
325
|
+
<ol>
|
|
326
|
+
<li><a href="#0-tldr">§0 TL;DR</a>
|
|
327
|
+
</li>
|
|
328
|
+
<li><a href="#1-基本设定--直觉">§1 基本设定 & 直觉</a>
|
|
329
|
+
</li>
|
|
330
|
+
<li><a href="#2-flow-matching-loss">§2 Flow Matching Loss</a>
|
|
331
|
+
<ul>
|
|
332
|
+
<li><a href="#21-marginal-flow-matching理论上">2.1 Marginal Flow Matching(理论上)</a></li>
|
|
333
|
+
<li><a href="#22-conditional-flow-matching实际可训练">2.2 Conditional Flow Matching(实际可训练)</a></li>
|
|
334
|
+
<li><a href="#23-关键定理lipman-et-al-2023-theorem-2">2.3 关键定理(Lipman et al. 2023, Theorem 2)</a></li>
|
|
335
|
+
</ul>
|
|
336
|
+
</li>
|
|
337
|
+
<li><a href="#3-三种-conditional-path-选择">§3 三种 Conditional Path 选择</a>
|
|
338
|
+
<ul>
|
|
339
|
+
<li><a href="#31-rectified-flow最简最稳最常用">3.1 Rectified Flow:最简、最稳、最常用</a></li>
|
|
340
|
+
<li><a href="#32-vp-path与-ddpm-同族">3.2 VP path(与 DDPM 同族)</a></li>
|
|
341
|
+
<li><a href="#33-ve-path与-smldedm-同族">3.3 VE path(与 SMLD/EDM 同族)</a></li>
|
|
342
|
+
</ul>
|
|
343
|
+
</li>
|
|
344
|
+
<li><a href="#4-训练代码框架pytorch">§4 训练代码框架(PyTorch)</a>
|
|
345
|
+
<ul>
|
|
346
|
+
<li><a href="#41-probability-path-抽象">4.1 Probability Path 抽象</a></li>
|
|
347
|
+
<li><a href="#42-vector-field-网络教学版-mlp实际用-u-net--dit">4.2 Vector Field 网络(教学版 MLP;实际用 U-Net / DiT)</a></li>
|
|
348
|
+
<li><a href="#43-cfm-loss">4.3 CFM Loss</a></li>
|
|
349
|
+
<li><a href="#44-minimal-training-loop">4.4 Minimal Training Loop</a></li>
|
|
350
|
+
</ul>
|
|
351
|
+
</li>
|
|
352
|
+
<li><a href="#5-ode-采样">§5 ODE 采样</a>
|
|
353
|
+
</li>
|
|
354
|
+
<li><a href="#6-与-diffusion--score-matching-的关系">§6 与 Diffusion / Score Matching 的关系</a>
|
|
355
|
+
<ul>
|
|
356
|
+
<li><a href="#61-velocity--score--noise-prediction-互换必考">6.1 Velocity ↔ Score ↔ Noise prediction 互换(必考)</a></li>
|
|
357
|
+
<li><a href="#62-几种-path-与-diffusion-的对应表">6.2 几种 path 与 diffusion 的对应表</a></li>
|
|
358
|
+
<li><a href="#63-为什么-rectified-flow-训练--采样相对稳">6.3 为什么 Rectified Flow 训练 / 采样相对"稳"</a></li>
|
|
359
|
+
</ul>
|
|
360
|
+
</li>
|
|
361
|
+
<li><a href="#7-高级话题">§7 高级话题</a>
|
|
362
|
+
<ul>
|
|
363
|
+
<li><a href="#71-reflowliu-et-al-2022-iclr">7.1 Reflow(Liu et al. 2022, ICLR)</a></li>
|
|
364
|
+
<li><a href="#72-conditional-flow-matching-cfg">7.2 Conditional Flow Matching (CFG)</a></li>
|
|
365
|
+
<li><a href="#73-logit-normal-tsd3-默认">7.3 Logit-normal $t$(SD3 默认)</a></li>
|
|
366
|
+
</ul>
|
|
367
|
+
</li>
|
|
368
|
+
<li><a href="#8-完整可运行示例2d-toy">§8 完整可运行示例(2D toy)</a>
|
|
369
|
+
</li>
|
|
370
|
+
</ol>
|
|
371
|
+
</nav>
|
|
372
|
+
|
|
373
|
+
<main>
|
|
374
|
+
<header class="hero">
|
|
375
|
+
<div class="eyebrow">Interview Prep · Flow Matching / Diffusion Generative Modeling</div>
|
|
376
|
+
<h1>Flow Matching Quick Reference</h1>
|
|
377
|
+
<p class="subtitle">Conditional Flow Matching · Rectified Flow / VP / VE · 训练 + 采样 + SD3 / FLUX 实战</p>
|
|
378
|
+
<p class="byline">By <strong>Ruofeng Yang (杨若峰), Shanghai Jiao Tong University</strong></p>
|
|
379
|
+
<div class="meta">
|
|
380
|
+
<span><strong>Source:</strong> <code>docs/tutorials/flow_matching_tutorial.md</code></span>
|
|
381
|
+
<span><strong>SHA256:</strong> <code>ab4358901d20</code></span>
|
|
382
|
+
<span><strong>Rendered:</strong> 2026-05-19 03:40 UTC</span>
|
|
383
|
+
|
|
384
|
+
</div>
|
|
385
|
+
</header>
|
|
386
|
+
|
|
387
|
+
<h2 id="0-tldr">§0 TL;DR</h2>
|
|
388
|
+
<div class="callout callout-info"><div class="callout-title">5 句话搞定 Flow Matching</div><p>一页拿下核心要点(详见后文 §1–§4 推导)。</p></div>
|
|
389
|
+
<ol><li><strong>目标</strong>:学一个 vector field $v_\theta(t, x)$,使 ODE $\dot{x}_t = v_\theta(t, x_t)$ 把 $x_0 \sim p_0$(噪声)演化到 $x_1 \sim p_1$(数据)。</li><li><strong>训练 (CFM)</strong>:$\mathcal{L}_\text{CFM}(\theta) = \mathbb{E}_{t, z, x_t \sim p_t(\cdot|z)} \|v_\theta(t, x_t) - u_t(x_t|z)\|^2$,<strong>simulation-free</strong>(不用解 ODE 算 loss)。</li><li><strong>关键定理</strong>:$\nabla_\theta \mathcal{L}_\text{FM} = \nabla_\theta \mathcal{L}_\text{CFM}$——所以学 conditional vector field 等价于学 marginal 的(Lipman et al. 2023)。</li><li><strong>最简版 (Rectified Flow / OT-CFM)</strong>:$x_t = (1-t)x_0 + tx_1$,target $u_t = x_1 - x_0$。SD3 / FLUX / Lumina 都用这个。</li><li><strong>采样</strong>:从 $x_0 \sim p_0$ 出发,用 ODE solver (Euler / Heun / RK4) 积分到 $t=1$。</li></ol>
|
|
390
|
+
<h2 id="1-基本设定--直觉">§1 基本设定 & 直觉</h2>
|
|
391
|
+
<p>给定数据分布 $p_1$("目标")和简单先验 $p_0$(一般 $\mathcal{N}(0, I)$),我们想构造一族<strong>概率路径</strong> $\{p_t\}_{t \in [0,1]}$ 从 $p_0$ 平滑过渡到 $p_1$。</p>
|
|
392
|
+
<div class="callout callout-warn"><div class="callout-title">Convention(全文统一)</div><p>后续公式记号见下表。</p></div>
|
|
393
|
+
<ul><li>$x_0 \sim p_0 = \mathcal{N}(0, I)$(噪声端)—— $t=0$</li><li>$x_1 \sim p_1$(数据端)—— $t=1$</li><li>采样方向:从 $t=0$ 积分到 $t=1$(noise → data)</li><li>注:不同论文 convention 不同——Lipman et al. 2023 用 $x_0$=data, $x_1$=noise;Liu et al. 2022 (Rectified Flow) 用 $x_0$=noise, $x_1$=data(本文采用)。SD3 论文也是 noise→data 但记号略有差异。<strong>面试时第一句先 disambiguate</strong>。</li></ul>
|
|
394
|
+
<p>一族<strong>时变 vector field</strong> $u_t : [0,1] \times \mathbb{R}^d \to \mathbb{R}^d$ 通过 ODE $\dot{x}_t = u_t(x_t)$ 把粒子从 $p_0$ 推到 $p_1$。由<strong>连续性方程</strong>:</p>
|
|
395
|
+
<p>$$\boxed{\;\frac{\partial p_t}{\partial t} + \nabla \cdot (p_t\, u_t) = 0\;}$$</p>
|
|
396
|
+
<p>我们的目标:找一个神经网络 $v_\theta(t, x) \approx u_t(x)$。</p>
|
|
397
|
+
<pre class="diagram"><code>
|
|
398
|
+
p_0 (噪声) p_t (中间) p_1 (数据)
|
|
399
|
+
●●●●● → ● ● ● → ████
|
|
400
|
+
v_θ(t, x)
|
|
401
|
+
─────────→
|
|
402
|
+
dx/dt = v_θ</code></pre>
|
|
403
|
+
<p>对比 diffusion:</p>
|
|
404
|
+
<ul><li><strong>Diffusion (SDE)</strong>:$dx = f(x, t) dt + g(t) dW$,训练用 score matching $s_\theta \approx \nabla \log p_t$</li><li><strong>Flow matching (ODE)</strong>:$dx = v_\theta(t, x) dt$,<strong>无随机项</strong>,训练直接回归 vector field</li><li>两者通过 <strong>probability flow ODE</strong> 关联:$v = f - \frac{1}{2} g^2 \nabla \log p_t$(见 §6)</li></ul>
|
|
405
|
+
<h2 id="2-flow-matching-loss">§2 Flow Matching Loss</h2>
|
|
406
|
+
<h3 id="21-marginal-flow-matching理论上">2.1 Marginal Flow Matching(理论上)</h3>
|
|
407
|
+
<p>如果我们知道 $u_t$(marginal vector field),直接回归即可:</p>
|
|
408
|
+
<p>$$\mathcal{L}_\text{FM}(\theta) = \mathbb{E}_{t \sim \mathcal{U}[0,1],\; x \sim p_t} \left\| v_\theta(t, x) - u_t(x) \right\|^2$$</p>
|
|
409
|
+
<p><strong>问题</strong>:$u_t(x)$ 是 marginal,由所有 conditional path 加权(积分),<strong>无法直接采样</strong>。</p>
|
|
410
|
+
<h3 id="22-conditional-flow-matching实际可训练">2.2 Conditional Flow Matching(实际可训练)</h3>
|
|
411
|
+
<p>引入 <strong>conditioning variable</strong> $z$(如 $z = x_1$,或 $z = (x_0, x_1)$)。选 conditional path $p_t(x | z)$ 和 conditional vector field $u_t(x | z)$,使得对 $z$ 边缘后等于 marginal:</p>
|
|
412
|
+
<p>$$p_t(x) = \int p_t(x | z) q(z)\, dz, \quad u_t(x) = \int u_t(x|z) \frac{p_t(x|z) q(z)}{p_t(x)} dz$$</p>
|
|
413
|
+
<p>那么 <strong>Conditional FM loss</strong>:</p>
|
|
414
|
+
<p>$$\boxed{\;\mathcal{L}_\text{CFM}(\theta) = \mathbb{E}_{t,\; z \sim q,\; x \sim p_t(\cdot|z)} \left\| v_\theta(t, x) - u_t(x|z) \right\|^2\;}$$</p>
|
|
415
|
+
<p>每一项都<strong>可采样、可计算</strong>。$x \sim p_t(\cdot|z)$ 一般是闭式可采样(如下面的线性插值)。</p>
|
|
416
|
+
<h3 id="23-关键定理lipman-et-al-2023-theorem-2">2.3 关键定理(Lipman et al. 2023, Theorem 2)</h3>
|
|
417
|
+
<div class="callout callout-good"><div class="callout-title">梯度等价定理</div><p>在 $p_t$ 与 $u_t$ 适当正则、$p_t > 0$ 等假设下:</p></div>
|
|
418
|
+
<p>$$\nabla_\theta \mathcal{L}_\text{FM}(\theta) = \nabla_\theta \mathcal{L}_\text{CFM}(\theta)$$ 所以 <strong>minimize CFM ≡ minimize FM</strong>。两个 loss 在以上前提下相差一个不依赖 $\theta$ 的常数。</p>
|
|
419
|
+
<p><strong>证明草图</strong>:展开二范数 $\|v_\theta\|^2 - 2 v_\theta^\top u_t + \|u_t\|^2$,前两项在两种 loss 下相等(用 $u_t$ 的定义把 marginal 写成 conditional 的加权期望),第三项不依赖 $\theta$,求梯度时消掉。</p>
|
|
420
|
+
<div class="callout callout-info"><div class="callout-title">面试加分:marginal vector field 非唯一</div><p>给定 $p_t$,满足连续性方程 $\partial_t p_t + \nabla\cdot(p_t u_t) = 0$ 的 $u_t$ <strong>不唯一</strong>——任意 divergence-free 向量场加上去仍是合法的。CFM 通过 conditional path 自动选了一个"自然的" $u_t$(通常对应 OT map 或 score-based ODE)。这点常被追问 "marginal $u_t$ 唯一吗?"。</p></div>
|
|
421
|
+
<h2 id="3-三种-conditional-path-选择">§3 三种 Conditional Path 选择</h2>
|
|
422
|
+
<p>conditioning 取 $z = (x_0, x_1)$,$x_0 \sim p_0$, $x_1 \sim p_1$。conditional path $p_t(x | x_0, x_1)$ 一般是 Dirac $\delta(x - \psi_t(x_0, x_1))$(确定性插值),conditional vector field 为 $\dot{\psi}_t(x_0, x_1)$。</p>
|
|
423
|
+
<table><thead><tr><th>Path</th><th>$x_t = \psi_t(x_0, x_1)$</th><th>Target $u_t$</th><th>用在哪</th></tr></thead><tbody><tr><td><strong>Rectified Flow / OT-CFM</strong></td><td>$(1-t)x_0 + t\, x_1$</td><td>$x_1 - x_0$(常数)</td><td>SD3, FLUX, Lumina, MovieGen</td></tr><tr><td><strong>VP cosine</strong></td><td>$\cos\!\left(\frac{\pi t}{2}\right) x_0 + \sin\!\left(\frac{\pi t}{2}\right) x_1$</td><td>$-\frac{\pi}{2}\sin\!\frac{\pi t}{2}\, x_0 + \frac{\pi}{2}\cos\!\frac{\pi t}{2}\, x_1$</td><td>与 DDPM cosine schedule 同族(在限制下)</td></tr><tr><td><strong>VE</strong></td><td>$x_1 + \sigma(1{-}t)\, x_0$,$\sigma$ 递增</td><td>$-\sigma'(1{-}t)\, x_0$</td><td>与 SMLD/EDM 同族(需 prior 方差匹配 $\sigma_{\max}^2$)</td></tr></tbody></table>
|
|
424
|
+
<h3 id="31-rectified-flow最简最稳最常用">3.1 Rectified Flow:最简、最稳、最常用</h3>
|
|
425
|
+
<p>线性插值:$x_t = (1-t) x_0 + t\, x_1$,所以 $\dot{x}_t = x_1 - x_0$ 是<strong>常数</strong>(不依赖 $t$)。</p>
|
|
426
|
+
<p>训练目标:</p>
|
|
427
|
+
<p>$$\mathcal{L}_\text{RF}(\theta) = \mathbb{E}_{t, x_0, x_1} \|v_\theta(t,\, (1-t)x_0 + t x_1) - (x_1 - x_0)\|^2$$</p>
|
|
428
|
+
<p>"OT-CFM" 名字来自:如果 $(x_0, x_1)$ 是 optimal transport coupling(不是独立采样),则学到的 vector field 实现近似 OT map。</p>
|
|
429
|
+
<div class="callout callout-good"><div class="callout-title">Reflow:Rectified Flow 的杀手锏</div><p>用学到的 $v_\theta$ 重新生成 $(x_0, x_1)$ pair(用 $x_0$ 跑 ODE 得到对应 $x_1$),<strong>再训一次</strong>。新的 trajectory 更直,<strong>少步数采样质量大幅提升</strong>,可以做到 1-step / 2-step 生成(InstaFlow 等)。</p></div>
|
|
430
|
+
<h3 id="32-vp-path与-ddpm-同族">3.2 VP path(与 DDPM 同族)</h3>
|
|
431
|
+
<p>用 $\sigma(t) = \cos\!\frac{\pi t}{2}$(噪声系数), $\alpha(t) = \sin\!\frac{\pi t}{2}$(数据系数),满足 $\sigma^2 + \alpha^2 = 1$(variance preserving):</p>
|
|
432
|
+
<p>$$x_t = \sigma(t)\, x_0 + \alpha(t)\, x_1, \quad u_t = \sigma'(t)\, x_0 + \alpha'(t)\, x_1$$</p>
|
|
433
|
+
<p>边界:$t=0$ 时 $x_t = x_0$(噪声),$t=1$ 时 $x_t = x_1$(数据)。</p>
|
|
434
|
+
<p>这种 path 和 DDPM 的 cosine schedule <strong>属于同一族 Gaussian path</strong>(连续极限 + 时间反向)。但严格意义上不能说"完全等价"——DDPM 原文 (Nichol-Dhariwal) 还有 $s=0.008$ offset 等细节,且 DDPM 是 forward noising convention($t=0$ 数据),FM 是反向($t=0$ 噪声)。</p>
|
|
435
|
+
<h3 id="33-ve-path与-smldedm-同族">3.3 VE path(与 SMLD/EDM 同族)</h3>
|
|
436
|
+
<p>沿用 Lipman et al. 2023 的 conditional VE path:</p>
|
|
437
|
+
<p>$$p_t(x | x_1) = \mathcal{N}\!\left(x \,\Big|\, x_1,\; \sigma(1-t)^2 I\right)$$</p>
|
|
438
|
+
<p>$\sigma(s)$ 在 forward time $s \in [0, 1]$ 上单调递增(如 $\sigma(s) = \sigma_\min (\sigma_\max/\sigma_\min)^s$)。reparameterize 得</p>
|
|
439
|
+
<p>$$x_t = x_1 + \sigma(1-t)\, x_0, \quad u_t = -\sigma'(1-t)\, x_0$$</p>
|
|
440
|
+
<p>边界:$t=0$ 时 $x_t \approx x_1 + \sigma_\max\, x_0$(大噪声主导),$t=1$ 时 $x_t \approx x_1 + \sigma_\min\, x_0 \approx x_1$(数据)。</p>
|
|
441
|
+
<div class="callout callout-warn"><div class="callout-title">VE 部署注意</div><p>prior $p_0$ 严格意义上应是 $\mathcal{N}(0, \sigma_\max^2 I)$(让 $t=0$ 的边缘方差匹配);用 $\mathcal{N}(0, I)$ 时要相应缩放(如 $x_0 \leftarrow \sigma_\max \cdot \tilde{x}_0$)。本文代码示例以教学为主,<strong>实战 VE 用 EDM preconditioning</strong> 更稳。</p></div>
|
|
442
|
+
<h2 id="4-训练代码框架pytorch">§4 训练代码框架(PyTorch)</h2>
|
|
443
|
+
<h3 id="41-probability-path-抽象">4.1 Probability Path 抽象</h3>
|
|
444
|
+
<pre><code class="language-python">import math
|
|
445
|
+
from dataclasses import dataclass
|
|
446
|
+
from typing import Callable, Optional
|
|
447
|
+
import torch
|
|
448
|
+
import torch.nn as nn
|
|
449
|
+
import torch.nn.functional as F
|
|
450
|
+
|
|
451
|
+
@dataclass
|
|
452
|
+
class FlowPath:
|
|
453
|
+
""" Conditional probability path 抽象 """
|
|
454
|
+
name: str
|
|
455
|
+
sample_xt: Callable # (t, x0, x1) -> x_t
|
|
456
|
+
target_ut: Callable # (t, x0, x1) -> u_t
|
|
457
|
+
|
|
458
|
+
def _broadcast_t(t: torch.Tensor, x: torch.Tensor) -> torch.Tensor:
|
|
459
|
+
""" t: [B], x: [B, ...] —— 把 t 扩成可广播 x 的 shape [B, 1, 1, ...] """
|
|
460
|
+
return t.view(-1, *([1] * (x.dim() - 1)))
|
|
461
|
+
|
|
462
|
+
def rectified_flow_path() -> FlowPath:
|
|
463
|
+
""" x_t = (1-t)x_0 + t*x_1, u_t = x_1 - x_0 """
|
|
464
|
+
def sample_xt(t, x0, x1):
|
|
465
|
+
tb = _broadcast_t(t, x0)
|
|
466
|
+
return (1 - tb) * x0 + tb * x1
|
|
467
|
+
def target_ut(t, x0, x1):
|
|
468
|
+
return x1 - x0
|
|
469
|
+
return FlowPath("rectified_flow", sample_xt, target_ut)
|
|
470
|
+
|
|
471
|
+
def vp_cosine_path() -> FlowPath:
|
|
472
|
+
""" x_t = cos(π t/2) x_0 + sin(π t/2) x_1
|
|
473
|
+
t=0: x_t = x_0 (noise); t=1: x_t = x_1 (data) [noise → data 方向] """
|
|
474
|
+
def sample_xt(t, x0, x1):
|
|
475
|
+
tb = _broadcast_t(t, x0)
|
|
476
|
+
sig = torch.cos(0.5 * math.pi * tb) # noise coeff
|
|
477
|
+
alp = torch.sin(0.5 * math.pi * tb) # data coeff
|
|
478
|
+
return sig * x0 + alp * x1
|
|
479
|
+
def target_ut(t, x0, x1):
|
|
480
|
+
tb = _broadcast_t(t, x0)
|
|
481
|
+
d_sig = -0.5 * math.pi * torch.sin(0.5 * math.pi * tb)
|
|
482
|
+
d_alp = 0.5 * math.pi * torch.cos(0.5 * math.pi * tb)
|
|
483
|
+
return d_sig * x0 + d_alp * x1
|
|
484
|
+
return FlowPath("vp_cosine", sample_xt, target_ut)
|
|
485
|
+
|
|
486
|
+
def ve_path(sigma_min: float = 0.01, sigma_max: float = 50.0) -> FlowPath:
|
|
487
|
+
""" VE: x_t = x_1 + σ(1-t) · x_0, σ(s) 在 forward time s 上递增 (log-linear)
|
|
488
|
+
t=0: x_t = x_1 + σ_max·x_0 (大噪声); t=1: x_t ≈ x_1 (数据)
|
|
489
|
+
注: 严格 VE 需 prior p_0 ~ N(0, σ_max² I),本示例为简化用 N(0, I),
|
|
490
|
+
实战需要 EDM-style preconditioning。 """
|
|
491
|
+
log_min, log_max = math.log(sigma_min), math.log(sigma_max)
|
|
492
|
+
def sigma_fwd(s): # 在 forward time s 上递增
|
|
493
|
+
return torch.exp(log_min * (1 - s) + log_max * s)
|
|
494
|
+
def d_sigma_fwd(s): # dσ/ds = σ · (log σ_max − log σ_min)
|
|
495
|
+
return sigma_fwd(s) * (log_max - log_min)
|
|
496
|
+
def sample_xt(t, x0, x1):
|
|
497
|
+
tb = _broadcast_t(t, x0)
|
|
498
|
+
return x1 + sigma_fwd(1 - tb) * x0
|
|
499
|
+
def target_ut(t, x0, x1):
|
|
500
|
+
tb = _broadcast_t(t, x0)
|
|
501
|
+
# u_t = d/dt [σ(1-t)] x_0 = -σ'(1-t) · x_0
|
|
502
|
+
return -d_sigma_fwd(1 - tb) * x0
|
|
503
|
+
return FlowPath("ve", sample_xt, target_ut)</code></pre>
|
|
504
|
+
<h3 id="42-vector-field-网络教学版-mlp实际用-u-net--dit">4.2 Vector Field 网络(教学版 MLP;实际用 U-Net / DiT)</h3>
|
|
505
|
+
<pre><code class="language-python">class SinusoidalTimeEmbed(nn.Module):
|
|
506
|
+
""" 与 Transformer positional embedding 同构的时间编码 """
|
|
507
|
+
def __init__(self, dim: int):
|
|
508
|
+
super().__init__()
|
|
509
|
+
self.dim = dim
|
|
510
|
+
def forward(self, t: torch.Tensor) -> torch.Tensor:
|
|
511
|
+
# t: [B] in [0, 1]
|
|
512
|
+
half = self.dim // 2
|
|
513
|
+
freqs = torch.exp(-math.log(10000) * torch.arange(half, device=t.device) / half)
|
|
514
|
+
args = t[:, None] * freqs[None, :]
|
|
515
|
+
return torch.cat([torch.sin(args), torch.cos(args)], dim=-1)
|
|
516
|
+
|
|
517
|
+
class VectorFieldMLP(nn.Module):
|
|
518
|
+
""" v_θ(t, x) ——— 简化版,用于 2D toy / low-dim 实验
|
|
519
|
+
实际生成模型把这里换成 U-Net (image) 或 DiT (high-res / video) """
|
|
520
|
+
def __init__(self, dim: int, hidden: int = 256, t_dim: int = 128):
|
|
521
|
+
super().__init__()
|
|
522
|
+
self.t_embed = nn.Sequential(
|
|
523
|
+
SinusoidalTimeEmbed(t_dim),
|
|
524
|
+
nn.Linear(t_dim, hidden),
|
|
525
|
+
nn.SiLU(),
|
|
526
|
+
nn.Linear(hidden, hidden),
|
|
527
|
+
)
|
|
528
|
+
self.net = nn.Sequential(
|
|
529
|
+
nn.Linear(dim + hidden, hidden), nn.SiLU(),
|
|
530
|
+
nn.Linear(hidden, hidden), nn.SiLU(),
|
|
531
|
+
nn.Linear(hidden, dim),
|
|
532
|
+
)
|
|
533
|
+
def forward(self, t: torch.Tensor, x: torch.Tensor) -> torch.Tensor:
|
|
534
|
+
# t: [B], x: [B, dim]
|
|
535
|
+
return self.net(torch.cat([x, self.t_embed(t)], dim=-1))</code></pre>
|
|
536
|
+
<h3 id="43-cfm-loss">4.3 CFM Loss</h3>
|
|
537
|
+
<pre><code class="language-python">def cfm_loss(
|
|
538
|
+
model: nn.Module,
|
|
539
|
+
path: FlowPath,
|
|
540
|
+
x1: torch.Tensor, # [B, ...] 数据样本
|
|
541
|
+
x0: Optional[torch.Tensor] = None, # 默认 N(0, I)
|
|
542
|
+
t_dist: str = "uniform", # "uniform" or "logitnormal"
|
|
543
|
+
return_components: bool = False,
|
|
544
|
+
):
|
|
545
|
+
"""
|
|
546
|
+
Conditional Flow Matching loss:
|
|
547
|
+
L = E ‖v_θ(t, x_t) - u_t(x_t | x_0, x_1)‖²
|
|
548
|
+
"""
|
|
549
|
+
B = x1.shape[0]
|
|
550
|
+
device = x1.device
|
|
551
|
+
if x0 is None:
|
|
552
|
+
x0 = torch.randn_like(x1)
|
|
553
|
+
|
|
554
|
+
# t 采样
|
|
555
|
+
if t_dist == "uniform":
|
|
556
|
+
t = torch.rand(B, device=device)
|
|
557
|
+
elif t_dist == "logitnormal":
|
|
558
|
+
# SD3 默认: t = σ(z), z ~ N(0, 1). 更集中在 t≈0.5(最难学的中间区)
|
|
559
|
+
t = torch.sigmoid(torch.randn(B, device=device))
|
|
560
|
+
else:
|
|
561
|
+
raise ValueError(f"unknown t_dist: {t_dist}")
|
|
562
|
+
|
|
563
|
+
x_t = path.sample_xt(t, x0, x1)
|
|
564
|
+
u_t = path.target_ut(t, x0, x1)
|
|
565
|
+
v_pred = model(t, x_t)
|
|
566
|
+
|
|
567
|
+
loss = F.mse_loss(v_pred, u_t)
|
|
568
|
+
if return_components:
|
|
569
|
+
return loss, {"v_pred_norm": v_pred.norm().item(), "u_norm": u_t.norm().item()}
|
|
570
|
+
return loss</code></pre>
|
|
571
|
+
<h3 id="44-minimal-training-loop">4.4 Minimal Training Loop</h3>
|
|
572
|
+
<pre><code class="language-python">def train_flow_matching(
|
|
573
|
+
model: nn.Module,
|
|
574
|
+
dataloader, # yields x1 batches
|
|
575
|
+
path: FlowPath,
|
|
576
|
+
total_steps: int = 50_000,
|
|
577
|
+
lr: float = 3e-4,
|
|
578
|
+
weight_decay: float = 0.0,
|
|
579
|
+
device: str = "cuda",
|
|
580
|
+
log_every: int = 200,
|
|
581
|
+
ema_decay: float = 0.9999, # EMA 对生成模型必加
|
|
582
|
+
):
|
|
583
|
+
model = model.to(device).train()
|
|
584
|
+
opt = torch.optim.AdamW(model.parameters(), lr=lr, weight_decay=weight_decay)
|
|
585
|
+
ema_model = _make_ema(model) # 见下
|
|
586
|
+
|
|
587
|
+
step = 0
|
|
588
|
+
while step < total_steps:
|
|
589
|
+
for x1 in dataloader:
|
|
590
|
+
x1 = x1.to(device, non_blocking=True)
|
|
591
|
+
loss = cfm_loss(model, path, x1, t_dist="logitnormal")
|
|
592
|
+
opt.zero_grad(set_to_none=True)
|
|
593
|
+
loss.backward()
|
|
594
|
+
torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0)
|
|
595
|
+
opt.step()
|
|
596
|
+
_update_ema(ema_model, model, ema_decay)
|
|
597
|
+
|
|
598
|
+
if step % log_every == 0:
|
|
599
|
+
print(f"[{step:6d}] {path.name} loss = {loss.item():.4f}")
|
|
600
|
+
step += 1
|
|
601
|
+
if step >= total_steps: break
|
|
602
|
+
|
|
603
|
+
return model, ema_model
|
|
604
|
+
|
|
605
|
+
@torch.no_grad()
|
|
606
|
+
def _make_ema(model):
|
|
607
|
+
import copy
|
|
608
|
+
ema = copy.deepcopy(model).eval()
|
|
609
|
+
for p in ema.parameters(): p.requires_grad_(False)
|
|
610
|
+
return ema
|
|
611
|
+
|
|
612
|
+
@torch.no_grad()
|
|
613
|
+
def _update_ema(ema, model, decay):
|
|
614
|
+
for ep, p in zip(ema.parameters(), model.parameters()):
|
|
615
|
+
ep.mul_(decay).add_(p.detach(), alpha=1 - decay)</code></pre>
|
|
616
|
+
<h2 id="5-ode-采样">§5 ODE 采样</h2>
|
|
617
|
+
<p>训练完 $v_\theta$ 后,从 $x_0 \sim p_0$ 出发,解 ODE $\dot{x}_t = v_\theta(t, x_t)$ 到 $t = 1$。</p>
|
|
618
|
+
<pre><code class="language-python">@torch.no_grad()
|
|
619
|
+
def euler_sampler(model, x0, steps=50, t_start=0.0, t_end=1.0):
|
|
620
|
+
""" 一阶 Euler:每步 1 NFE,简单但需要较多步数 """
|
|
621
|
+
x = x0.clone()
|
|
622
|
+
ts = torch.linspace(t_start, t_end, steps + 1, device=x0.device)
|
|
623
|
+
for i in range(steps):
|
|
624
|
+
t = ts[i].expand(x.shape[0])
|
|
625
|
+
dt = ts[i + 1] - ts[i]
|
|
626
|
+
x = x + dt * model(t, x)
|
|
627
|
+
return x
|
|
628
|
+
|
|
629
|
+
@torch.no_grad()
|
|
630
|
+
def heun_sampler(model, x0, steps=50, t_start=0.0, t_end=1.0):
|
|
631
|
+
""" 二阶 Heun (improved Euler / RK2):每步 2 NFE,精度 O(dt²) """
|
|
632
|
+
x = x0.clone()
|
|
633
|
+
ts = torch.linspace(t_start, t_end, steps + 1, device=x0.device)
|
|
634
|
+
for i in range(steps):
|
|
635
|
+
b = x.shape[0]
|
|
636
|
+
t_i, t_next = ts[i], ts[i + 1]
|
|
637
|
+
dt = t_next - t_i
|
|
638
|
+
v1 = model(t_i.expand(b), x)
|
|
639
|
+
x_euler = x + dt * v1
|
|
640
|
+
v2 = model(t_next.expand(b), x_euler)
|
|
641
|
+
x = x + dt * 0.5 * (v1 + v2)
|
|
642
|
+
return x
|
|
643
|
+
|
|
644
|
+
@torch.no_grad()
|
|
645
|
+
def rk4_sampler(model, x0, steps=25, t_start=0.0, t_end=1.0):
|
|
646
|
+
""" 四阶 Runge-Kutta:每步 4 NFE,精度 O(dt⁴)
|
|
647
|
+
25 步 × 4 NFE = 100 NFE,但通常比 100 步 Euler 准很多 """
|
|
648
|
+
x = x0.clone()
|
|
649
|
+
ts = torch.linspace(t_start, t_end, steps + 1, device=x0.device)
|
|
650
|
+
for i in range(steps):
|
|
651
|
+
b = x.shape[0]
|
|
652
|
+
t_i, t_next = ts[i], ts[i + 1]
|
|
653
|
+
dt = t_next - t_i
|
|
654
|
+
k1 = model(t_i.expand(b), x)
|
|
655
|
+
k2 = model((t_i + dt / 2).expand(b), x + dt / 2 * k1)
|
|
656
|
+
k3 = model((t_i + dt / 2).expand(b), x + dt / 2 * k2)
|
|
657
|
+
k4 = model(t_next.expand(b), x + dt * k3)
|
|
658
|
+
x = x + dt / 6 * (k1 + 2 * k2 + 2 * k3 + k4)
|
|
659
|
+
return x</code></pre>
|
|
660
|
+
<div class="callout callout-info"><div class="callout-title">Sampler 选择 cheat sheet</div><p>按 NFE / 质量 trade-off 排序如下。</p></div>
|
|
661
|
+
<ul><li><strong>Euler</strong>:1 NFE/step,需 ≥50 步才出好图;调试 baseline 用</li><li><strong>Heun / RK2</strong>:2 NFE/step,~25 步质量已不错;EDM 默认</li><li><strong>RK4</strong>:4 NFE/step,10-20 步通常等同 Euler 100 步</li><li><strong>Adaptive (Dopri5 / dopri8)</strong>:torchdiffeq 提供;自动控制误差,但 NFE 不可控</li><li><strong>Rectified Flow 重训后</strong>:经 1-2 次 reflow,1-4 步 Euler 即可达到接近多步质量</li></ul>
|
|
662
|
+
<h2 id="6-与-diffusion--score-matching-的关系">§6 与 Diffusion / Score Matching 的关系</h2>
|
|
663
|
+
<p>对任意 SDE $dx = f(x, t) dt + g(t) dW$(forward),存在对应的 <strong>probability flow ODE</strong>(Song et al. 2021):</p>
|
|
664
|
+
<p>$$dx = \underbrace{\left[ f(x, t) - \frac{1}{2} g^2(t)\, \nabla_x \log p_t(x) \right]}_{\text{vector field } u_t(x)} dt$$</p>
|
|
665
|
+
<p>这个 ODE 在每一时刻的边缘分布 $p_t$ 与 SDE 完全一样。</p>
|
|
666
|
+
<div class="callout callout-good"><div class="callout-title">FM ↔ Score Matching 的桥梁(带前提)</div><p>当 FM 的概率路径源自一个非退化的 noising SDE ($g(t) > 0$) 时,学 score $s_\theta \approx \nabla \log p_t$ 与学 vector field $v_\theta \approx u_t$ 是<strong>同一信息的两种参数化</strong>:</p></div>
|
|
667
|
+
<p>$$v_\theta(t, x) = f(x, t) - \tfrac{1}{2} g^2(t)\, s_\theta(t, x)$$ 所以在 VP/VE path 下,FM 可被视为 score matching 在 ODE 视角下的等价参数化。<strong>但对 Rectified Flow / OT-CFM 不成立</strong>(没有标准 SDE 对应),那里 FM 是更一般的 vector-field regression。</p>
|
|
668
|
+
<h3 id="61-velocity--score--noise-prediction-互换必考">6.1 Velocity ↔ Score ↔ Noise prediction 互换(必考)</h3>
|
|
669
|
+
<p>VP/VE path 下,假设 $x_t = \alpha(t) x_1 + \sigma(t) x_0$($x_0 \sim \mathcal{N}(0, I)$ 是噪声方向),三种主流 prediction target 之间是线性可逆转换:</p>
|
|
670
|
+
<p>$$ \begin{aligned} \epsilon\text{-prediction} &:\quad \epsilon_\theta(t, x_t) \approx x_0 \\ x_0\text{-prediction} &:\quad x^0_\theta(t, x_t) \approx x_1 \\ v\text{-prediction (Salimans-Ho)} &:\quad v_\theta(t, x_t) \approx \alpha'(t) x_1 + \sigma'(t) x_0 \\ \text{score} &:\quad s_\theta(t, x_t) \approx -x_0 / \sigma(t) \end{aligned} $$</p>
|
|
671
|
+
<p>已知 $x_t$ 和任一 prediction,可代数恢复其他三种。例如 VP 下 $\epsilon$ 与 score 关系:</p>
|
|
672
|
+
<p>$$s_\theta(t, x_t) = -\epsilon_\theta(t, x_t) / \sigma(t)$$</p>
|
|
673
|
+
<p>这就是为什么 DDPM (学 $\epsilon$) 与 score-based (学 $\nabla \log p_t$) 是<strong>等价参数化</strong>。Flow matching 学 $v = \alpha' x_1 + \sigma' x_0$ 也是其中一种,且在 RF (linear) 下退化为 $v = x_1 - x_0$。</p>
|
|
674
|
+
<h3 id="62-几种-path-与-diffusion-的对应表">6.2 几种 path 与 diffusion 的对应表</h3>
|
|
675
|
+
<table><thead><tr><th>FM Path</th><th>等价 Diffusion / SDE</th><th>典型 noise schedule</th></tr></thead><tbody><tr><td>VP cosine</td><td>DDPM (cosine)</td><td>$\bar\alpha_t = \cos^2(\pi t/2)$</td></tr><tr><td>VP linear</td><td>DDPM (linear β)</td><td>$\beta_t = \beta_0 + t(\beta_1 - \beta_0)$</td></tr><tr><td>VE</td><td>SMLD / EDM</td><td>$\sigma_t \in [\sigma_\min, \sigma_\max]$ 对数线性</td></tr><tr><td>Rectified Flow</td><td>没有标准非零扩散 noising SDE 对应(退化情形除外)</td><td>路径本身是直线,"最短" path</td></tr></tbody></table>
|
|
676
|
+
<h3 id="63-为什么-rectified-flow-训练--采样相对稳">6.3 为什么 Rectified Flow 训练 / 采样相对"稳"</h3>
|
|
677
|
+
<ul><li><strong>常数 target</strong>:$u_t = x_1 - x_0$ 不显式依赖 $t$(在给定 $x_0, x_1$ 后),数值上易拟合</li><li><strong>直线 path</strong>:少步数 ODE 积分误差小</li><li><strong>Loss conditioning</strong>:RF 本身比 native DDPM 训练更平衡;但<strong>不是说不需要 reweighting</strong>——SD3 在 RF 之上仍做 logit-normal $t$ sampling 等 reweighting 并 ablate 出涨点</li><li><strong>Reflow 可压缩 NFE</strong>:1-step 生成路线(InstaFlow / SD3-Turbo / Flux-Schnell)</li></ul>
|
|
678
|
+
<h2 id="7-高级话题">§7 高级话题</h2>
|
|
679
|
+
<h3 id="71-reflowliu-et-al-2022-iclr">7.1 Reflow(Liu et al. 2022, ICLR)</h3>
|
|
680
|
+
<p>Rectified Flow 之所以能少步数生成,关键是 <strong>reflow 算法</strong>:</p>
|
|
681
|
+
<ol><li>第一次训练得到 $v_\theta^{(1)}$(用独立配对 $(x_0, x_1) \sim p_0 \otimes p_1$)</li><li>用 $v_\theta^{(1)}$ 跑 ODE 生成<strong>coupled</strong> pair $(x_0, x_1^{(1)})$,即 $x_1^{(1)} = \text{ODE}(x_0; v_\theta^{(1)})$</li><li>用 coupled pair 重新训练得到 $v_\theta^{(2)}$,新的 trajectory 更<strong>直</strong></li><li>重复——Liu et al. 2022 证明在适当假设下,配对的 <strong>convex transport cost</strong> 非增(每次 reflow 不会让总传输成本变差)</li></ol>
|
|
682
|
+
<p>"trajectory 变直"是直觉与经验观察;具体严格定理是 transport cost 单调性。实际 1-2 次 reflow 就能做到 4-step 媲美 50-step(InstaFlow / SD3-Turbo / Flux-Schnell)。极限:完全直线 → 1-step 生成($x_1 = x_0 + v_\theta(0, x_0)$)。</p>
|
|
683
|
+
<h3 id="72-conditional-flow-matching-cfg">7.2 Conditional Flow Matching (CFG)</h3>
|
|
684
|
+
<p>对条件生成(如 text-to-image),模型接收额外条件 $c$:</p>
|
|
685
|
+
<p>$$v_\theta(t, x, c)$$</p>
|
|
686
|
+
<p>训练时以概率 $p_\text{drop}$(一般 0.1)把 $c$ 替换成空(如 null embedding),得到 <strong>无条件 head</strong>。</p>
|
|
687
|
+
<p>采样时用 <strong>Classifier-Free Guidance</strong>:</p>
|
|
688
|
+
<p>$$v_\text{CFG}(t, x, c) = v_\theta(t, x, \emptyset) + s \cdot \left[v_\theta(t, x, c) - v_\theta(t, x, \emptyset)\right]$$</p>
|
|
689
|
+
<p>$s$ 是 guidance scale(一般 1.5-7.5)。$s > 1$ 时放大 conditional 信号,提升文本对齐但损失多样性。</p>
|
|
690
|
+
<h3 id="73-logit-normal-tsd3-默认">7.3 Logit-normal $t$(SD3 默认)</h3>
|
|
691
|
+
<p>SD3 (Esser et al. 2024) 发现,<strong>$t \sim \mathcal{U}[0, 1]$ 不是最优</strong>。中间区域($t \approx 0.5$)的 target 噪声-信号比最难学。改成:</p>
|
|
692
|
+
<p>$$t = \sigma(\tau), \quad \tau \sim \mathcal{N}(m, s^2)$$</p>
|
|
693
|
+
<p>即 $\tau$ 高斯采样后 sigmoid 映射回 $(0, 1)$,可调 $m, s$ 控制 $t$ 分布偏重哪段。默认 $m = 0, s = 1$ 时 $t$ 集中在 0.5 附近。这是 SD3 论文中 ablation 涨点的关键之一。</p>
|
|
694
|
+
<h2 id="8-完整可运行示例2d-toy">§8 完整可运行示例(2D toy)</h2>
|
|
695
|
+
<p>下面是端到端最小可运行示例:训练 vector field 学习把 $\mathcal{N}(0, I)$ 映到一个 2D 月亮形分布。</p>
|
|
696
|
+
<pre><code class="language-python">if __name__ == "__main__":
|
|
697
|
+
# 1) 数据 (target distribution p_1): 2D 月亮形
|
|
698
|
+
from sklearn.datasets import make_moons
|
|
699
|
+
|
|
700
|
+
def sample_moons(n: int) -> torch.Tensor:
|
|
701
|
+
X, _ = make_moons(n_samples=n, noise=0.05)
|
|
702
|
+
return torch.tensor(X, dtype=torch.float32) * 2.0 # scale
|
|
703
|
+
|
|
704
|
+
# 2) 模型 + path
|
|
705
|
+
model = VectorFieldMLP(dim=2, hidden=128)
|
|
706
|
+
path = rectified_flow_path()
|
|
707
|
+
|
|
708
|
+
# 3) "dataloader" (随机生成)
|
|
709
|
+
class MoonDataset:
|
|
710
|
+
def __init__(self, batch=512, total=5000):
|
|
711
|
+
self.batch = batch; self.total = total
|
|
712
|
+
def __iter__(self):
|
|
713
|
+
for _ in range(self.total):
|
|
714
|
+
yield sample_moons(self.batch)
|
|
715
|
+
|
|
716
|
+
# 4) 训练
|
|
717
|
+
train_flow_matching(
|
|
718
|
+
model,
|
|
719
|
+
MoonDataset(batch=512, total=2000),
|
|
720
|
+
path=path,
|
|
721
|
+
total_steps=2000,
|
|
722
|
+
lr=3e-4,
|
|
723
|
+
device="cuda" if torch.cuda.is_available() else "cpu",
|
|
724
|
+
log_every=100,
|
|
725
|
+
)
|
|
726
|
+
|
|
727
|
+
# 5) 采样
|
|
728
|
+
model.eval()
|
|
729
|
+
device = next(model.parameters()).device
|
|
730
|
+
x0 = torch.randn(2000, 2, device=device)
|
|
731
|
+
x_samples = euler_sampler(model, x0, steps=50)
|
|
732
|
+
|
|
733
|
+
# 与真实 2D 月亮叠图,可视化检验
|
|
734
|
+
import matplotlib.pyplot as plt
|
|
735
|
+
real = sample_moons(2000).numpy()
|
|
736
|
+
fake = x_samples.cpu().numpy()
|
|
737
|
+
plt.scatter(real[:, 0], real[:, 1], alpha=0.3, label="real")
|
|
738
|
+
plt.scatter(fake[:, 0], fake[:, 1], alpha=0.3, label="generated")
|
|
739
|
+
plt.legend(); plt.savefig("flow_matching_moons.png", dpi=120)</code></pre>
|
|
740
|
+
<div class="callout callout-warn"><div class="callout-title">Production 化要补的(教学版未含)</div><p>上线前需补的工程项如下。</p></div>
|
|
741
|
+
<ul><li><strong>EMA scheduler</strong>:训练后期 decay 更接近 1(如 0.9999 → 0.99995)</li><li><strong>Gradient checkpointing</strong>:U-Net / DiT 显存优化</li><li><strong>Mixed precision</strong>:fp16 / bf16 + GradScaler</li><li><strong>Latent space</strong>:高分辨率图像在 VAE latent 里跑 FM(LDM / SD3 / FLUX)</li><li><strong>Conditioning</strong>:text encoder (T5 / CLIP) + cross-attention 或 token concat</li><li><strong>Distributed</strong>:DDP / FSDP for multi-GPU</li><li><strong>Loss weighting</strong>:SD3 用 logit-normal $t$ 已隐式 reweighting;EDM 显式 SNR weighting</li></ul>
|
|
742
|
+
<p><strong>Flow Matching Quick Reference</strong> · 主要参考:Lipman et al. 2023 (Flow Matching), Liu et al. 2022 (Rectified Flow), Esser et al. 2024 (SD3 / MM-DiT)</p>
|
|
743
|
+
|
|
744
|
+
<footer class="aris-footer">
|
|
745
|
+
Generated by <a href="https://github.com/wanshuiyin/Auto-claude-code-research-in-sleep/blob/main/skills/render-html/SKILL.md">ARIS <code>/render-html</code></a> ·
|
|
746
|
+
source path <code>docs/tutorials/flow_matching_tutorial.md</code> ·
|
|
747
|
+
SHA256 <code>ab4358901d20</code> ·
|
|
748
|
+
generated at 2026-05-19 03:40 UTC.
|
|
749
|
+
This is a generated view — edit the source Markdown, then re-render.
|
|
750
|
+
</footer>
|
|
751
|
+
</main>
|
|
752
|
+
|
|
753
|
+
</div>
|
|
754
|
+
</body>
|
|
755
|
+
</html>
|