Kuiper cleanup 5/n - Remove kuiper_calibration_per_segment
#164
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
flinder
requested review from
Lorenzo-Perini,
TaXxER and
danielha1
as code owners
meta-cla
bot
added
the
CLA Signed
flinder
force-pushed
the
export-D91299869
branch
from
03c5e46 to
abc3300
Compare
flinder
force-pushed
the
export-D91299869
branch
from
abc3300 to
2871f5b
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 23, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
2871f5b to
030cc91
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 23, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
030cc91 to
5dd7cc0
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 23, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 23, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
5dd7cc0 to
1abac99
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
1abac99 to
febf007
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
febf007 to
8cb5fa0
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
8cb5fa0 to
4bd678e
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
…lue (facebookincubator#160) Summary: Introduces a cleaner public API for Expected Cumulative Calibration Error (ECCE) metrics. The new ecce(), ecce_sigma(), and ecce_pvalue() functions provide simplified interfaces to the existing kuiper calibration functionality, making the API more intuitive for users who want to compute calibration errors without needing to understand the underlying Kuiper test implementation details. Deprecation of old versions of the functions will come in follow up diffs. ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: TaXxER Differential Revision: D91235672
Summary: Removes the public `kuiper_calibration` function and updates internal callers to use the new ECCE API or lower-level functions directly. ### Changes **Removed:** - `kuiper_calibration()` - wrapper function that computed Kuiper calibration distance **Refactored:** - `kuiper_test()` - now calls `ecce_sigma()` directly instead of `kuiper_calibration(..., normalization_method="kuiper_standard_deviation")` [Note: this function will be removed in a future commit] - `ecce()` - now calls `kuiper_calibration_per_segment(...).item()` directly instead of going through `kuiper_calibration()` - `ecce_sigma()` - now calls `kuiper_calibration_per_segment(...).item()` directly instead of going through `kuiper_calibration()` ### Rationale This is part of a cleanup series migrating from the legacy "kuiper" naming to the cleaner "ECCE" (Expected Cumulative Calibration Error) API. The `kuiper_calibration` function was just a thin wrapper that called `kuiper_calibration_per_segment().item()`, so callers now use that pattern directly. ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91235674
) Summary: * Remove the function `kuiper_distribution` and replace with private helper `_ecce_cdf` * Add public helper `ecce_pvalue_from_sigma()` to compute p-value from sigma-scaled ECCE statistic, encapsulating the min/max checking logic * Rename constants `KUIPER_STATISTIC_MIN/MAX` to `ECCE_SIGMA_MIN/MAX` * Refactor all internal usages (`kuiper_test`, `ecce_pvalue`, `MulticalibrationError.p_value`, `MulticalibrationError.segment_p_values`) to use the new helpers * Update external call site in swisscheese `mce.py` to use `ecce_pvalue_from_sigma` ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91298776
Summary: Removes the deprecated kuiper_pvalue function and migrates all usages to the new ecce_pvalue API. ## Changes **Removed from metrics.py:** * kuiper_pvalue() function (callers should use ecce_pvalue() instead) **Updated external call sites**: * fbcode/fblearner/flow/projects/fluent2/workflows/evaluation/metrics/kuiper_calibration_pvalue.py: Renamed _compute_kuiper_pvalue() → _compute_ecce_pvalue() Now calls multicalibration_metrics.ecce_pvalue() instead of kuiper_pvalue() Updated tests: * test_metrics.py: Renamed test_kuiper_pvalue_wrapper_returns_pvalue_only → test_ecce_pvalue_wrapper_returns_pvalue_only and updated to use metrics.ecce_pvalue() ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91298790
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
force-pushed
the
export-D91299869
branch
from
4bd678e to
af458d8
Compare
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
flinder
added a commit
to flinder/MCGrad-1
that referenced
this pull request
Jan 26, 2026
…kincubator#164) Summary: Removes the public `kuiper_calibration_per_segment` function and replaces it with a simplified private helper `_ecce_per_segment` that only computes unnormalized ECCE. Changes: - Replace `kuiper_calibration_per_segment` with `_ecce_per_segment` (removed normalization parameter since it was always None in the only call site) - Make `ecce()` implement its own logic directly using `_calculate_cumulative_differences` - Make `ecce_sigma()` call `ecce()` and divide by `kuiper_standard_deviation` with proper zero-handling - Update `MulticalibrationError.segment_ecces_absolute` to use the new `_ecce_per_segment` helper ------------------------------------- Target API design ------------------------------------- ``` ┌─────────────────────────────────────────────────────────────────────────────────┐ │ ECCE API ARCHITECTURE │ ├─────────────────────────────────────────────────────────────────────────────────┤ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ PUBLIC API (User-Facing) ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ ┌─────────────┐ ┌──────────────┐ ┌──────────────────────────┐ ║ │ │ ║ │ ecce() │ │ ecce_sigma() │ │ ecce_pvalue() │ ║ │ │ ║ │ │ │ │ │ │ ║ │ │ ║ │ Unnormalized│ │ Normalized │ │ P-value for │ ║ │ │ ║ │ calibration │───▶│ by std dev │───▶│ calibration test │ ║ │ │ ║ │ error │ │ (sigma) │ │ │ ║ │ │ ║ └──────┬──────┘ └──────┬───────┘ └────────────┬─────────────┘ ║ │ │ ║ │ │ │ ║ │ │ ╚══════════╪══════════════════╪═════════════════════════╪══════════════════╝ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌──────────────────────────┐ │ │ │ │ │ ecce_pvalue_from_sigma()│ │ │ │ │ │ (public helper) │ │ │ │ │ └────────────┬─────────────┘ │ │ │ │ │ │ │ ┌──────────┴──────────────────┴────────────────────────┴───────────────────┐ │ │ │ PRIVATE HELPERS │ │ │ ├──────────────────────────────────────────────────────────────────────────┤ │ │ │ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _calculate_cumulative_ │ │ _ecce_cdf() │ │ │ │ │ │ differences() │ │ │ │ │ │ │ │ │ │ CDF for Brownian motion │ │ │ │ │ │ Core computation of weighted │ │ range (for p-value calc) │ │ │ │ │ │ cumulative differences │ │ │ │ │ │ │ └───────────────┬────────────────┘ └─────────────────────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌────────────────────────────────┐ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_per_segment() │ │ _ecce_standard_deviation() │ │ │ │ │ │ │ │ │ │ │ │ │ │ Vectorized ECCE computation │ │ Wrapper → calls per_segment│ │ │ │ │ │ for multiple segments │ │ version with .item() │ │ │ │ │ └────────────────────────────────┘ └──────────────┬──────────────┘ │ │ │ │ │ │ │ │ │ ▼ │ │ │ │ ┌─────────────────────────────┐ │ │ │ │ │ _ecce_standard_deviation_ │ │ │ │ │ │ per_segment() │ │ │ │ │ │ │ │ │ │ │ │ Variance-based std dev │ │ │ │ │ │ for multiple segments │ │ │ │ │ └─────────────────────────────┘ │ │ │ │ │ │ │ └──────────────────────────────────────────────────────────────────────────┘ │ │ │ │ ╔═══════════════════════════════════════════════════════════════════════════╗ │ │ ║ MulticalibrationError Class ║ │ │ ╠═══════════════════════════════════════════════════════════════════════════╣ │ │ ║ ║ │ │ ║ Computes calibration metrics across multiple data segments ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Cached Properties │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_ecces_absolute ──────────▶ _ecce_per_segment() │ ║ │ │ ║ │ (ECCE per segment) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_sigmas ──────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (std dev per segment) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ sigma_0 ─────────────────────────▶ _ecce_standard_deviation_ │ ║ │ │ ║ │ (global std dev) per_segment() │ ║ │ │ ║ │ │ ║ │ │ ║ │ p_value ─────────────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (min p-value across segments) │ ║ │ │ ║ │ │ ║ │ │ ║ │ segment_p_values ────────────────▶ ecce_pvalue_from_sigma() │ ║ │ │ ║ │ (p-value per segment) (vectorized) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ║ ┌─────────────────────────────────────────────────────────────────────┐ ║ │ │ ║ │ Derived Metrics │ ║ │ │ ║ ├─────────────────────────────────────────────────────────────────────┤ ║ │ │ ║ │ │ ║ │ │ ║ │ mce = max(segment_ecces_absolute) │ ║ │ │ ║ │ mce_sigma_scale = max(segment_ecces_absolute / segment_sigmas) │ ║ │ │ ║ │ mce_relative = mce / ecce_0 (normalized by global ECCE) │ ║ │ │ ║ │ │ ║ │ │ ║ └─────────────────────────────────────────────────────────────────────┘ ║ │ │ ║ ║ │ │ ╚═══════════════════════════════════════════════════════════════════════════╝ │ │ │ └─────────────────────────────────────────────────────────────────────────────────┘ ``` Reviewed By: Lorenzo-Perini Differential Revision: D91299869
|
This pull request has been merged in 9641aab. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary:
Removes the public
kuiper_calibration_per_segmentfunction and replaces it with a simplified private helper_ecce_per_segmentthat only computes unnormalized ECCE.Changes:
kuiper_calibration_per_segmentwith_ecce_per_segment(removed normalization parameter since it was always None in the only call site)ecce()implement its own logic directly using_calculate_cumulative_differencesecce_sigma()callecce()and divide bykuiper_standard_deviationwith proper zero-handlingMulticalibrationError.segment_ecces_absoluteto use the new_ecce_per_segmenthelperTarget API design
Differential Revision: D91299869