rpotter6298
708fbc70ce
- Introduced `bridge_attention_ceiling_check.py` for variance decomposition analysis on bridge attention configurations. - Added `bridge_attention_readout.py` to perform per-tower gate and contribution readouts, including AUC sanity checks. - Created multiple JSON configuration files for backbone replication experiments, including anonymous CV variants and basic backbones. - Implemented sensitivity experiments to evaluate the impact of axial length inclusion and EfficientNetV2-M performance at higher resolutions. - Added a memory probe script to assess GPU memory usage during training with EfficientNetV2-M.
124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
"""F8 quadrant attention chart (full-image scope).
|
|
|
|
Reads the per-cell quadrant fractions produced by
|
|
``v4.figures.F8_explainability`` (see CSV at
|
|
``output/F8_quadrant_fractions.csv``) and renders a single-panel grouped bar
|
|
chart of full-image Grad-CAM intensity by optic-disc quadrant. Within-disc
|
|
and peri-disc scopes are present in the CSV but are not plotted here; see
|
|
the CSV for the per-cell numbers if you need them.
|
|
|
|
Re-run:
|
|
python -m v4.figures.F8_quadrant_plot
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import matplotlib
|
|
matplotlib.use("Agg")
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import pandas as pd
|
|
|
|
|
|
CSV = Path(__file__).parent / "output" / "F8_quadrant_fractions.csv"
|
|
OUT = Path(__file__).parent / "output" / "F8_quadrant_attention.png"
|
|
|
|
QUAD_ORDER = ("ST", "SN", "IT", "IN")
|
|
QUAD_LABEL = {
|
|
"ST": "Superotemporal",
|
|
"SN": "Superonasal",
|
|
"IT": "Inferotemporal",
|
|
"IN": "Inferonasal",
|
|
}
|
|
|
|
CELL_ORDER = [
|
|
("Normal", "correct"),
|
|
("Normal", "incorrect"),
|
|
("Glaucoma", "correct"),
|
|
("Glaucoma", "incorrect"),
|
|
]
|
|
CELL_COLOR = {
|
|
("Normal", "correct"): "#3B6FB5",
|
|
("Normal", "incorrect"): "#8BB0DA",
|
|
("Glaucoma", "correct"): "#c44e52",
|
|
("Glaucoma", "incorrect"): "#e6a3a4",
|
|
}
|
|
CELL_LABEL = {
|
|
("Normal", "correct"): "Normal correct",
|
|
("Normal", "incorrect"): "Normal incorrect",
|
|
("Glaucoma", "correct"): "Glaucoma correct",
|
|
("Glaucoma", "incorrect"): "Glaucoma incorrect",
|
|
}
|
|
|
|
|
|
def render() -> None:
|
|
if not CSV.exists():
|
|
raise SystemExit(
|
|
f"CSV {CSV} not found. Run `python -m v4.figures.F8_explainability "
|
|
"--only-gradcam --run-gradcam` first."
|
|
)
|
|
df = pd.read_csv(CSV)
|
|
df_full = df[df["scope"] == "full"]
|
|
|
|
fig, ax = plt.subplots(figsize=(9.6, 6.0))
|
|
|
|
n_quad = len(QUAD_ORDER)
|
|
n_cell = len(CELL_ORDER)
|
|
bar_w = 0.18
|
|
x = np.arange(n_quad, dtype=float)
|
|
|
|
for i, cell in enumerate(CELL_ORDER):
|
|
means, sds = [], []
|
|
n_eyes = None
|
|
for q in QUAD_ORDER:
|
|
row = df_full[
|
|
(df_full["class"] == cell[0])
|
|
& (df_full["outcome"] == cell[1])
|
|
& (df_full["quadrant"] == q)
|
|
]
|
|
means.append(float(row["mean"].iloc[0]) if len(row) else float("nan"))
|
|
sds.append(float(row["sd"].iloc[0]) if len(row) else float("nan"))
|
|
if n_eyes is None and len(row):
|
|
n_eyes = int(row["n_eyes"].iloc[0])
|
|
|
|
offsets = (i - (n_cell - 1) / 2.0) * bar_w
|
|
bars = ax.bar(
|
|
x + offsets, means, width=bar_w,
|
|
yerr=sds, capsize=2,
|
|
color=CELL_COLOR[cell], alpha=0.92,
|
|
edgecolor="black", linewidth=0.6,
|
|
label=f"{CELL_LABEL[cell]} (n = {n_eyes})",
|
|
error_kw=dict(ecolor="#444", linewidth=0.8, capthick=0.8),
|
|
)
|
|
for rect, m in zip(bars, means):
|
|
if np.isnan(m):
|
|
continue
|
|
ax.text(rect.get_x() + rect.get_width() / 2, m + 0.005,
|
|
f"{m:.2f}", ha="center", va="bottom",
|
|
fontsize=8.5, color="#222")
|
|
|
|
ax.set_xticks(x)
|
|
ax.set_xticklabels([QUAD_LABEL[q] for q in QUAD_ORDER], fontsize=11)
|
|
ax.set_ylabel("Mean fraction of full-image Grad-CAM intensity", fontsize=11)
|
|
ax.set_title(
|
|
"Image-tower Grad-CAM by optic-disc quadrant (OD-oriented; centroid-split)",
|
|
fontsize=12.5, fontweight="bold",
|
|
)
|
|
ax.grid(axis="y", alpha=0.3, linestyle="--")
|
|
ax.set_ylim(0, max(ax.get_ylim()[1], 0.7))
|
|
ax.axhline(0.25, color="#888", linestyle=":", linewidth=0.8, alpha=0.6, zorder=0)
|
|
|
|
ax.legend(loc="upper center", bbox_to_anchor=(0.5, -0.10), ncol=4,
|
|
framealpha=0.94, fontsize=10)
|
|
|
|
fig.tight_layout(rect=(0, 0.04, 1, 0.98))
|
|
OUT.parent.mkdir(parents=True, exist_ok=True)
|
|
fig.savefig(OUT, dpi=180, bbox_inches="tight")
|
|
plt.close(fig)
|
|
print(f"saved {OUT}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
render()
|