Restructure: descriptive tier and experiment names, paper/manuscript

- paper/pnas -> paper/manuscript (venue-neutral)
- configs/layer1 -> configs/inheritance, src/knowledge -> src/inheritance
  (imported as `inheritance`), make layer1 -> make inheritance; layer2 alias dropped
- inheritance and trained-network bundles named after the manuscript figure
  they feed (fig2_grounding_sweep, figS3_rebaselining, ...), or descriptively
  where they feed none; configs keep their `experiment:` value so parquet
  hashes are unchanged, only output.dir moves
- figure scripts, SI figure sources, notebooks, REPRODUCING.md, README and the
  SI Methods/tables updated; make clean no longer deletes tracked manifests;
  reproduce.sh hashes the s{seed}/ layouts too

Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Y64o8FKP7rCuXzC48pxpMm
This commit is contained in:
Giorgio Gilestro 2026-09-13 17:00:40 +01:00
parent 84124de143
commit ab3dc10587
240 changed files with 477 additions and 476 deletions

View file

@ -0,0 +1,81 @@
"""`architectures` figure — the Wright-Fisher collapse operator is architecture-general.
The same dry-collapse / grounding-rescue signature appears in three distinct inductive biases:
the exact histogram (multinomial), an autoregressive GRU, and a causal-masked MLP. If the
signs held only for the histogram, the effect would be an artefact of the exact operator;
seeing them in every trained architecture is the generality claim.
Three panels: (A) forward-KL trajectories per architecture, dry (solid) vs grounded (dashed);
(B) stationary forward-KL, dry vs grounded, grouped by architecture (all fall with grounding);
(C) tail-item survival, dry vs grounded, grouped by architecture (all rise). Reads only the
committed bundle.
Usage: python figures/plot_figS1_architectures.py [results/figS1_architectures]
"""
from __future__ import annotations
import sys
from pathlib import Path
import matplotlib.pyplot as plt
import numpy as np
sys.path.insert(0, str(Path(__file__).parent))
from _figlib import load_bundle, mean_ci, savefig, letter_axes # noqa: E402
_ARCH_ORDER = ["histogram", "rnn", "mlp"]
_ARCH_LABEL = {"histogram": "histogram\n(exact)", "rnn": "GRU\n(autoregressive)",
"mlp": "MLP\n(causal-masked)"}
def main(results_dir: str = "results/figS1_architectures") -> None:
df, cfg = load_bundle(results_dir)
kinds = [k for k in _ARCH_ORDER if k in set(df["kind"].unique())]
g_dry, g_wet = min(df["g"].unique()), max(df["g"].unique())
last = int(cfg["generations"] * 0.6)
stat = df[df["generation"] >= last]
fig, axes = plt.subplots(1, 3, figsize=(16, 4.6))
arch_colors = dict(zip(kinds, plt.cm.tab10(np.arange(len(kinds)))))
# Panel A: forward-KL trajectories per architecture, dry (solid) vs grounded (dashed).
ax = axes[0]
for k in kinds:
for g, ls, alpha in [(g_dry, "-", 1.0), (g_wet, "--", 0.7)]:
s = df[(df["kind"] == k) & (df["g"] == g)].groupby("generation")["forward_kl"].mean()
ax.plot(s.index, s.values, ls, color=arch_colors[k], alpha=alpha, lw=1.8,
label=f"{k} (g={g:g})")
ax.set(xlabel="generation", ylabel=r"forward-KL $D(p^*\Vert\hat p)$",
title="No real data (solid) collapses;\ngrounded (dashed) holds in every architecture")
ax.legend(frameon=False, fontsize=7, ncol=1)
# Panels B & C: grouped bars, dry vs grounded per architecture.
def grouped_bar(ax, metric, title, ylabel):
x = np.arange(len(kinds))
w = 0.36
for off, g, lab, col in [(-w / 2, g_dry, f"no real data (g={g_dry:g})", "#d62728"),
(w / 2, g_wet, f"grounded (g={g_wet:g})", "#2ca02c")]:
means, errs = [], []
for k in kinds:
sub = stat[(stat["kind"] == k) & (stat["g"] == g)]
_, m, ci = mean_ci(sub.assign(_x=0), "_x", metric)
means.append(m[0]); errs.append(ci[0])
ax.bar(x + off, means, w, yerr=errs, capsize=3, label=lab, color=col, alpha=0.85)
ax.set_xticks(x)
ax.set_xticklabels([_ARCH_LABEL[k] for k in kinds], fontsize=8)
ax.set(ylabel=ylabel, title=title)
ax.legend(frameon=False, fontsize=8)
grouped_bar(axes[1], "forward_kl", "Stationary forward-KL falls with grounding",
r"stationary forward-KL")
grouped_bar(axes[2], "tail_frac_alive", "Tail-item survival rises with grounding",
"tail items alive")
fig.tight_layout()
letter_axes(fig)
savefig(fig, results_dir, "figS1_architectures")
if __name__ == "__main__":
main(*sys.argv[1:])