REVIEW 3 major objections 4 minor 1 cited by
Learning Parametric Convex Functions
T0 review · 3 major / 4 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read A neural network architecture with nonnegative skip-layer weights and convex activations produces parametrized convex functions that can be fit to data and used directly in convex optimization.
desk verdict A clean, practical extension of ICNNs to parametric convex functions; the core proof is sound, but the quadratic experiment's sampling description and the monotonicity prose need fixing. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the parameter network $\psi(\theta)$, a hypernetwork that takes $\theta$ as input and emits all layer weights, input-skip weights, and offsets of the main network. Its output entries for the matrices $W^l(\theta)$, $l=2,\dots,L$, are passed through a nonnegative activation such as ReLU or logistic, which is what guarantees convexity in $x$ for every $\theta$. The convexity proof is the composition rule from convex analysis: a nondecreasing convex function composed with a nonnegative weighted sum of convex functions plus an affine term remains convex. This is the same rule that disciplined convex programming uses, which is why the fitted expression can be translated into a modeling-language expression for convex optimization.
What would settle it
Train the architecture on data from $f(x,\theta)=x^T Q(\theta)x$ with $Q(\theta)$ a positive semidefinite matrix whose entries oscillate at increasing frequency in $\theta$; if the test error grows without bound as the frequency increases while the training error stays small, the hypernetwork's expressiveness is insufficient for that family.
Extended reading notes
Core claim
The central claim is that the architecture defined by $z_0=x$, $z_l=\phi(W^l(\theta)z_{l-1}+V^l(\theta)x+\omega^l(\theta))$ for $l=1,\dots,L-1$, and $y=W^L(\theta)z_{L-1}+V^L(\theta)x+\omega^L(\theta)$, with all $W^l(\theta)$ elementwise nonnegative for $l=2,\dots,L$ and $\phi$ a nondecreasing convex activation, produces a parametrized convex function for any $\theta$. The argument is a recursion: nonnegative weighted sums of convex functions are convex, and the composition rule preserves convexity, exactly as in disciplined convex programming. Thus the model is a PCF by construction and is DCP-expressible. Fitting the network weights by minimizing a regularized average loss therefore yields a data-driven PCF that can be used in convex optimization formulations.
Load-bearing premise
The method assumes that the secondary network that generates the weights can represent the weight matrices accurately enough over the entire parameter range, with no guarantee that this is possible; success depends on the chosen architecture and on the optimizer finding a good fit.
Editorial extensions
If this is right
- A fitted PCF can be exported to a disciplined convex programming expression and placed anywhere a convex expression is allowed, such as an objective or a constraint, enabling parametrized convex optimization families built from data.
- The same architecture supports extensions: adding a convex quadratic term $x^T Q(\theta)x$, enforcing monotonicity in $x$, and encouraging a prescribed minimizer via a subgradient penalty.
- The framework also fits parametrized convex sets by learning a PCF level-set boundary $\{x : f(x,\theta)\le 0\}$ from membership data.
- In the battery aging example, the learned PCF achieves about seven times lower test root-mean-square error than the short-term convex approximation it is compared against.
- In approximate dynamic programming, the learned PCF serves as a convex surrogate value function, so the one-step lookahead control policy solves a convex problem at each time step.
Reading between the lines
- The paper does not state this, but the architecture's differentiability with respect to $x$ means the same training framework could also supervise the gradient of the PCF directly, which would be useful for value functions in control where gradient information is readily available.
- A testable extension is to apply the method to higher-dimensional variables and parameters; the current experiments are low-dimensional, and performance may degrade when the hypernetwork must interpolate many weight matrices across a large $\Theta$.
- The paper's expressiveness assumption points to a concrete research direction: proving function-approximation or sample-complexity bounds for this parameterized input-convex network class, which would tell practitioners how large the hypernetwork must be for a target accuracy.
- Another consequence not developed in the paper is that the fitted functions inherit the atom library of disciplined convex programming, so the same learned expression could be composed with existing atoms to build interpretable convex models without retraining.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper addresses the problem of fitting a parametrized convex function (PCF) to data, i.e., a function f(x, θ) that is convex in x for every allowed θ and is expressible in a disciplined convex programming (DCP) language. The proposed architecture (Eqs. 3–4) generalizes input-convex neural networks: the layer weights, offsets, and optional quadratic term are generated by a hypernetwork ψ(θ), and convexity is enforced by requiring the forward weight matrices W^l(θ) to be elementwise nonnegative and the activation to be nondecreasing and convex. The authors implement this in the open-source package LPCF, describe extensions for quadratic terms, monotonicity, subgradient specification, and convex-set fitting, and report experiments on a univariate piecewise-affine function, a quadratic function on R^3, a battery-aging model, and an approximate-dynamic-programming pendulum example. The central convexity argument is correct and DCP-compatible, and the paper offers a clean path from data to reusable convex expressions in CVXPY.
Significance. If the empirical claims hold, the paper provides a useful and practical tool: a data-driven way to obtain DCP-expressible PCFs that can be inserted directly into convex optimization families. The core mathematical construction in Section 2.1 is sound; the composition argument using nonnegative W^l and nondecreasing convex activations is exactly the standard DCP rule. The open-source implementation, the reproducible experimental pipeline, and the honest comparison with a nonparametric baseline in Section 4.1 are clear strengths. The main weakness is that the paper offers no expressiveness or sample-complexity analysis for the combined ICNN-plus-hypernetwork class, and one of the four experiments (Section 4.2) has a data-generation flaw that weakens the validation. These issues are fixable and do not undermine the convexity proof, but they do affect the strength of the paper's central practical claim.
major comments (3)
- [Section 4.2] The quadratic experiment is described as generating f_true(x, θ) = x^T θ x with θ ∈ S^3_+ (the positive semidefinite cone), which is convex in x. However, the experimental setup states that each entry of θ is drawn uniformly from [-1, 1]; a random symmetric matrix with entries in [-1, 1] is not generally positive semidefinite, so many of the sampled instances are not convex in x. The reported RMSE therefore does not specifically demonstrate learning a convex family; it conflates approximation error with the effect of projecting nonconvex data onto the convex class. Please either sample θ in a way that guarantees PSD (e.g., θ = U^T U with random entries in U) and report the convex subset, or explicitly reframe the experiment as fitting the best convex approximation and remove the PSD claim.
- [Section 2.2 and Section 4] The paper's central practical claim of a 'seamless path from data to a PCF' rests on the hypernetwork ψ being able to represent the weight matrices W^l(θ), V^l(θ), and offsets with sufficient accuracy over the parameter range Θ. Section 2.2 gives default hyperparameters and training heuristics, but provides no expressiveness result, no sample-complexity bound, and no capacity study for the combined ICNN-plus-hypernetwork architecture. If ψ is too small, or if the nonconvex fitting problem is not solved well, the learned f may be far from the target PCF. This is a genuine gap in the analysis; I ask the authors to either add a discussion of the approximation limitations or provide an ablation varying the widths of ψ and reporting train/test RMSE, or both.
- [Section 3.2] The monotone-decreasing construction is stated as 'cascading instead V^j with a nonpositive activation function φ−, such as φ− = −φ+', which as written suggests replacing the activation φ in Eq. (3) by −ReLU or −softplus. Such a replacement would destroy convexity, since −ReLU is not convex. The intended construction is presumably to make the entries of V^j nonpositive while keeping φ nondecreasing and convex, in which case the stated conclusion (convex and decreasing) is correct. Please rewrite this paragraph to remove the ambiguity and avoid presenting a false construction.
minor comments (4)
- [Section 1.4] In the paragraph on non-parametric PCF fitting, the phrase 'when p > 0 and r = 0' appears to be a typo; the preceding sentence already treats p > 0, and the regularizer r was not introduced in that context. Please clarify the intended condition.
- [Figure 2] The code snippet in Figure 2 contains a misspelling in the comment 'inv ol vi ng f' and in the variable name 'P ar am et er'; these should be corrected for readability.
- [Section 4.3] The experimental setup says 'We generate 1000 values of θ, with accumulated charge throughput A ∈ [0, 50] and temperature T ∈ [10, 50]' but then notes that Q = 1 is fixed. Since θ = (A, Q, T), the description should state explicitly that Q is not sampled.
- [References] Reference [Kin14] is titled 'ADAN', which appears to be a typo for 'ADAM'. Please correct it.
Circularity Check
No significant circularity: the convexity proof is self-contained and the fitting experiments use independently generated data.
full rationale
The paper's central claim is that the architecture (3)-(4), with elementwise nonnegative W^l(theta) for l=2,...,L and nondecreasing convex activations, yields a parametrized convex function. This is established in Section 2.1 by an induction argument using the standard composition rule for convex functions, cited to Boyd and Vandenberghe. The argument does not rely on any fitted parameter, on data, or on self-citation; it is a direct mathematical proof from the stated architecture assumptions. The fitting procedure in Section 1.4 is standard regularized empirical risk minimization, and the experiments in Section 4 evaluate the fitted PCF against independently generated data: a piecewise affine function, a quadratic function, a battery aging model, and a pendulum optimal control problem. Baselines such as the short-term battery approximation from [NOBL25] are external comparators, not premises of the convexity claim. The self-citations to prior work by the authors (e.g., [BAB20], [NOBL25]) provide background or baselines and are not load-bearing for the derivation. The limitation noted in Section 1.4 that the unregularized non-parametric solution is likely to generalize poorly is a caveat about that baseline method, not a circular step. The inconsistency in Section 4.2, where theta is said to lie in S^3_+ but entries are sampled uniformly in [-1,1], weakens that particular experiment's validity, but it is a data-generation bug rather than a fitted-input-called-prediction or self-definitional circularity. No step in the paper reduces to its own inputs by construction.
Assumptions & free parameters
free parameters (2)
- Regularization hyperparameter lambda =
0 by default; elastic net coefficients 10^-8 (quadratic) and 0.1 (L1) in Section 4.4
- Subgradient regularization weight rho_min =
not reported
assumptions (4)
- standard math Composition rule for convex functions: if phi is nondecreasing and convex, then phi composed with a convex function is convex.
- domain assumption The chosen activations (ReLU, softplus) are nondecreasing and convex.
- domain assumption The hypernetwork psi(theta) is sufficiently expressive and trainable to represent the required weight mappings.
- standard math Nonnegative linear combinations of convex functions, plus affine terms, are convex.
Cite this review
Pith. "Pith review of Learning Parametric Convex Functions." pith.science (2026). https://pith.science/paper/7XLD6UQJ
@misc{pith2026250604183,
author = {Pith},
title = {Pith review of: Learning Parametric Convex Functions},
year = {2026},
howpublished = {\url{https://pith.science/paper/7XLD6UQJ}},
note = {Machine review of arXiv:2506.04183}
}
read the original abstract
A parametrized convex function depends on a variable and a parameter, and is convex in the variable for any valid value of the parameter. Such functions can be used to specify parametrized convex optimization problems, i.e., a convex optimization family, in domain specific languages for convex optimization. In this paper we address the problem of fitting a parametrized convex function that is compatible with disciplined programming, to some given data. This allows us to fit a function arising in a convex optimization formulation directly to observed or simulated data. We demonstrate our open-source implementation on several examples, ranging from illustrative to practical.
Figures
Figures from the paper (3 more)
Forward citations
Cited by 1 Pith paper
-
Estimating Dynamic Marginal Policy Effects under Sequential Unconfoundedness
Parametric nonconvex problems can be approximated by a min of quasiconvex (convex∘monotonic) surrogates and solved via parallel convex optimization, trained mainly from function evaluations.
Reference graph
Works this paper leans on
-
[1]
Agrawal, R
A. Agrawal, R. Verschueren, S. Diamond, and S. Boyd. A rewriting system for convex optimization problems. Journal of Control and Decision , 5(1):42--60, 2018
2018
-
[2]
B. Amos, L. Xu, and Z. Kolter. Input convex neural networks. In International conference on machine learning , pages 146--155. PMLR, 2017
work page 2017
-
[3]
S. Abdufattokhov, M. Zanon, and A. Bemporad. Learning Lyapunov terminal costs from data for complexity reduction in nonlinear model predictive control. Int. J. Robust Nonlinear Control , 34(13):8676--8691, 2024
work page 2024
-
[4]
S. Boyd, A. Agrawal, and S. Barratt. Embedded convex optimization for control. In Proceedings 59th IEEE Conference on Decision and Control . IEEE, 2020
work page 2020
-
[5]
A. Bemporad. Linear and nonlinear system identification under _1 - and group- Lasso regularization via L-BFGS-B . Submitted for publication , 2024. Also available on arXiv at http://arxiv.org/abs/2403.03827
work page Pith review arXiv 2024
- [6]
-
[7]
J. Bradbury, R. Frostig, P. Hawkins, M. Johnson, C. Leary, D. Maclaurin, G. Necula, A. Paszke, J. Vander P las, S. Wanderman- M ilne, and Q. Zhang. JAX : C omposable transformations of P ython+ N um P y programs, 2018. Available at http://github.com/jax-ml/jax
work page 2018
-
[8]
R. Byrd, P. Lu, J. Nocedal, and C. Zhu. A limited memory algorithm for bound constrained optimization. SIAM Journal on Scientific Computing , 16(5):1190--1208, 1995
work page 1995
Show all 33 references
-
[9]
Boyd and L
S. Boyd and L. Vandenberghe. Convex Optimization . Cambridge University Press, 2004
2004
-
[10]
Diamond and S
S. Diamond and S. Boyd. CVXPY : A P ython-embedded modeling language for convex optimization. Journal of Machine Learning Research , 17(83):1--5, 2016
2016
-
[11]
Dunning, J
I. Dunning, J. Huchette, and M. Lubin. JuMP : A modeling language for mathematical optimization. SIAM Review , 59(2):295--320, 2017
2017
-
[12]
N. Draper. Applied regression analysis . McGraw-Hill. Inc, 1998
1998
-
[13]
Deschatre and X
T. Deschatre and X. Warin. Input convex K olmogorov A rnold networks. arXiv preprint arXiv:2505.21208 , 2025
2025
-
[14]
Ebbesen, P
S. Ebbesen, P. Elbert, and L. Guzzella. Battery state-of-health perceptive energy management for hybrid electric vehicles. IEEE Transactions on Vehicular technology , 61(7):2893--2900, 2012
2012
-
[15]
A. Fu, B. Narasimhan, and S. Boyd. CVXR : A n R package for disciplined convex optimization. Journal of Statistical Software , 94(14):1--34, 2020
2020
-
[16]
Grant and S
M. Grant and S. Boyd. CVX : M atlab software for disciplined convex programming, version 2.1, 2014
2014
-
[17]
Garcia, D
C. Garcia, D. Prett, and M. Morari. Model predictive control: T heory and practice -- a survey. Automatica , 25(3):335--348, 1989
1989
-
[18]
Huber and E
P. Huber and E. Ronchetti. Robust statistics . John Wiley & Sons, 2011
2011
-
[19]
P. Huber. Robust estimation of a location parameter. In Breakthroughs in statistics: M ethodology and distribution , pages 492--518. Springer, 1992
1992
-
[20]
K. He, X. Zhang, S. Ren, and J. Sun. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition , pages 770--778. IEEE, 2016
2016
-
[21]
K. He, X. Zhang, S. Ren, and J. Sun. Identity mappings in deep residual networks. In Computer Vision--ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, October 11--14, 2016, Proceedings, Part IV 14 , pages 630--645. Springer, 2016
2016
-
[22]
Kouvaritakis and M
B. Kouvaritakis and M. Cannon. Model predictive control . Springer, 2016
2016
-
[23]
D. Kingma. ADAN : A method for stochastic optimization. arXiv preprint arXiv:1412.6980 , 2014
2014 arXiv
-
[24]
L\"ofberg
J. L\"ofberg. YALMIP : A toolbox for modeling and optimization in Matlab . In IEEE International Conference on Robotics and Automation (ICRA) , pages 284--289. IEEE, 2004
2004
-
[25]
Liu and F
Y. Liu and F. Oliveira. ICNN -enhanced 2SP : Leveraging input convex neural networks for solving two-stage stochastic programming. arXiv preprint arXiv:2505.05261 , 2025
2025 arXiv
-
[26]
Nnorom, G
O. Nnorom, G. Ogut, S. Boyd, and P. Levis. Aging-aware battery control via convex optimization. arXiv preprint arXiv:2505.09030 , 2025
2025 arXiv
-
[27]
K. Pearson. On the general theory of skew correlation and non-linear regression . Dulau and Company, 1905
1905
-
[28]
Rawlings, D
J. Rawlings, D. Mayne, M. Diehl, et al. Model predictive control: T heory, computation, and design , volume 2. Nob Hill Publishing Madison, WI, 2017
2017
-
[29]
Suri and S
G. Suri and S. Onori. A control-oriented cycle-life model for hybrid electric vehicle lithium-ion batteries. Energy , 96:644--653, 2016
2016
-
[30]
Serrao, S
L. Serrao, S. Onori, A. Sciarretta, Y. Guezennec, and G. Rizzoni. Optimal energy management of hybrid electric vehicles including battery aging. In Proceedings of the 2011 American control conference , pages 2125--2130. IEEE, 2011
2011
-
[31]
Stellato
B. Stellato. Mixed-integer optimal control of fast dynamical systems . PhD thesis, University of Oxford, 2017
2017
-
[32]
Udell, K
M. Udell, K. Mohan, D. Zeng, J. Hong, S. Diamond, and S. Boyd. Convex optimization in J ulia. SC14 Workshop on High Performance Technical Computing in Dynamic Languages , 2014
2014
-
[33]
Zou and T
H. Zou and T. Hastie. Regularization and variable selection via the elastic net. Journal of the Royal Statistical Society Series B: S tatistical Methodology , 67(2):301--320, 2005
2005
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.