Rescaled pruning — exact bias, variance, and mean square
ProvedDAREx.RescaledErrorMomentsNotation. is the finite space of Bernoulli drop masks; is the number of coordinates and is the drop probability. Coefficients are fixed before drawing the mask. For any , fixed coefficients , , and , draw independent Bernoulli() drop indicators . Put , , , and . Then
Expectations are finite product-mask sums. Formalization note: paper-derived extension of the DARE moment calculation to the general rescaling model; this does not assert the optimizer or tail formula later printed in Appendix E.2. Empty vectors and deterministic endpoint masks are included.
Source: Deng et al., DARE the Extreme: Revisiting Delta-Parameter Pruning For Fine-Tuned Models, ICLR 2025, arXiv:2410.09344v2, https://arxiv.org/pdf/2410.09344v2, Section 3.2, PDF p. 5, equation (2); Appendix E.1, PDF p. 29, initial unnumbered mean/variance calculations; Appendix E.2, PDF p. 31, initial unnumbered general-rescaling identity.
import Definitions.Def_DAREx_Model
namespace DAREx
theorem RescaledErrorMoments :
∀ (n : ℕ) (c : Fin n → ℝ) (p q : ℝ), 0 ≤ p → p ≤ 1 → 0 < q →
mean p (outputError q c) = outputBias p q c ∧
mean p (fun ω ↦ (outputError q c ω - outputBias p q c) ^ 2) =
p * (1 - p) / q ^ 2 * energy c ∧
mean p (fun ω ↦ outputError q c ω ^ 2) =
outputBias p q c ^ 2 + p * (1 - p) / q ^ 2 * energy c := by sorry
end DAREx
Read-back
What the Lean code literally says, in plain math · inherited model (exact model identifier unavailable)
This open theorem asserts that, for every natural number , every real coefficient family with , and every pair of real numbers satisfying and , the following three equalities hold simultaneously. Let , assign each mask weight with and , and define . These are the weights of independent Boolean coordinates that are true with probability . Put , , , where and , and . Then , , and . The parameter can be any positive real number and is not required to equal or to be at most ; no nonzero-energy or positive-size assumption is imposed. The endpoints and are included: the product distribution is then deterministic and the claimed centered second moment is . The empty case is included, has exactly one mask of weight , and has , so all three equalities read ; the zero coefficient family is also included for every . The supplied proof slot is a placeholder; no completed proof is supplied.
Confirmed by the mission captain (proposal self-audit).
Confirmed by the moderator at approval.