AGENT SKILL

cdc_similarity_loss

Compute cross-domain distance consistency softmax-KL losses for few-shot generator adaptation experiments.

License 许可
Apache-2.0
最近更新
2026-09-03
Stars
★ 351
Forks
27

01 前言 Front Matter

name
cdc_similarity_loss
description
Compute cross-domain distance consistency softmax-KL losses for few-shot generator adaptation experiments.

02 资源文件 2

03 SKILL.md 正文

Cross-Domain Distance Consistency Loss

Use this skill when implementing or auditing a few-shot generator adaptation run that must preserve the relative relationships learned by a pretrained source generator. The skill is appropriate for full image features, intermediate generator activations, or reduced vector proxies. Do not use it as a target-realism loss by itself; it only checks whether source and adapted samples keep comparable pairwise similarity structure.

Inputs

  • source_layers: mapping from layer name to a list of source activation vectors for the same latent batch.
  • adapted_layers: mapping with identical layer names and vector counts for the adapted generator.
  • Optional temperature and eps values for stable probability and KL calculations.

Outputs

  • Total CDC loss averaged across layers and anchor rows.
  • Per-layer KL divergence values.
  • Source and adapted row-wise similarity distributions for debugging.

Workflow

  1. For each layer, compute cosine similarities between each anchor vector and every non-self vector in the batch.
  2. Convert each anchor row to a probability distribution using a temperature-scaled softmax.
  3. Compute KL divergence from the source distribution to the adapted distribution for every anchor row.
  4. Average the KL values across layers and rows and save row diagnostics when requested.
  5. Treat identical source/adapted activations as a near-zero-loss sanity check.

Validation

Run python tests/test_cdc_loss.py or validate this skill tree with validate_skill_tree.py --run-tests.

Limitations

This skill does not train a generator, compute LPIPS, or replace discriminator losses. In a reduced recovery it can operate on vectors, but the recovery must explicitly label that execution as proxy evidence.

04 同仓库更多 Skills

VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-triton-opt

Make a single PassNet Triton kernel fast AND numerically faithful: performance model (when a replacement can win at all), block/grid/warp tuning, autotune policy, launch overhead, and per-op numeric recipes to pass the dtype baseline tolerances. Use when a pass matches and is correct but speedup below expected, or when correctness fails by small numeric margins.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-solve

END-TO-END playbook for solving one PassNet sample: analyze the computation graphs, decide the optimization strategy (what to fuse, what to replace, what to leave alone), drive the iteration loop, and maximize the sample score. This is the ENTRY skill — invoke it first for any PassNet optimization task; it tells you when to use passnet-pattern-fusion, passnet-triton-opt and passnet-feedback.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-skill

PassNet GPU kernel optimization via compiler passes. Design and implement Triton-based optimization passes, create pass files under ./pass_dir/, self-evaluate with pass_evaluator, and iterate to maximize GPU speedup.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-pattern-fusion

Author PassNet pass files: write patterns that actually MATCH the FX graph, pick fusion regions (which ops to absorb into one Triton kernel), and structure multi-pass submissions with the mandatory shared-dispatch architecture. Use when creating or fixing pattern()/replacement_args()/replacement_func() files, when a pass "failed to match", or when deciding how to fuse multiple kernels/ops.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-orchestrate

Round/batch planning and decision support for PassNet graph optimization. Use for multi-sample triage, worker allocation, eval-budget policy, and post-eval keep/revert decisions. For solving one concrete sample end to end, use passnet-solve as the entry skill; consult this skill only when you need broader planning or a specific decision gate.

★ 351
VectorSpaceLab
VectorSpaceLab/AREX-Skill

passnet-feedback

Fast, deterministic feedback for PassNet work: pre-flight pattern/match verification WITHOUT burning a GPU evaluation, per-node bottleneck analysis of a sample's graphs, and eval-log parsing into per-variant status + estimated score + failure classification. Use BEFORE every GPU evaluation (check_pattern), at the START of a sample (analyze_graph), and AFTER every evaluation (parse_eval_log).

★ 351