Measure of Expectations Heterogeneity

You can easily deal with nonlinear transformations of expected values using auxiliary variables. See Expected value of a power
That makes your objective feasible.