hessboost 0.2.1

//! Regression objectives.

use super::{GradPair, Loss, check_label_domain, newton_intercepts, weighted_label_mean};
use crate::K_RT_EPS_F32;
use crate::data::MetaInfo;
use crate::error::Result;
use crate::metric::EvalMetric;
use crate::objective::PseudoHuber;

/// Squared-error regression (`reg:squarederror`), XGBoost's
/// `RegLossObj<LinearSquareLoss>`.
///
/// Loss `½ (pred − label)²` gives gradient `pred − label` and constant Hessian
/// `1`, both times the row weight, which `scale_pos_weight` multiplies for a
/// label of exactly `1`. The prediction transform is the identity and the
/// optimal base margin is the (weighted) label mean, or with
/// `scale_pos_weight != 1` XGBoost's Newton step (`FitIntercept`) on the
/// reweighted gradients.
#[derive(Debug, Clone, Copy)]
pub(crate) struct SquaredError {
    scale_pos_weight: f32,
}

impl SquaredError {
    /// The loss with positive-label weight `scale_pos_weight`.
    pub(crate) fn new(scale_pos_weight: f32) -> Self {
        SquaredError { scale_pos_weight }
    }
}

impl Default for SquaredError {
    fn default() -> Self {
        SquaredError::new(1.0)
    }
}

impl Loss for SquaredError {
    fn name(&self) -> &'static str {
        "reg:squarederror"
    }

    fn gradient(
        &self,
        preds: &[f32],
        labels: &[f32],
        weights: Option<&[f32]>,
        out: &mut [GradPair],
    ) {
        let scale_pos_weight = self.scale_pos_weight;
        super::elementwise_gradient(preds, labels, weights, out, |p, y, mut w| {
            // XGBoost `RegLossObj::GetGradient`: `w *= scale_pos_weight`
            // for a label of exactly 1, in `f32`.
            if y == 1.0 {
                w *= scale_pos_weight;
            }
            GradPair::new((p - y) * w, w)
        });
    }

    fn const_hess(&self) -> bool {
        // XGBoost's `LinearSquareLoss::Info` whatever `scale_pos_weight`:
        // the Hessian does not depend on the margin.
        true
    }

    fn base_margins_info(&self, info: &MetaInfo) -> Vec<f32> {
        // XGBoost `RegLossObj::InitEstimation`: `FitInterceptGlmLike`'s
        // (weighted) label mean, unless `scale_pos_weight` is in play, in
        // which case the reweighted loss needs the Newton step.
        if (self.scale_pos_weight - 1.0).abs() > K_RT_EPS_F32 {
            return newton_intercepts(self, info);
        }
        vec![weighted_label_mean(info.label_values(), info.weights)]
    }

    fn pointwise_loss(&self) -> Option<super::PointwiseLoss<'_>> {
        // `½ (margin − y)²`, with a label of 1 reweighted by
        // `scale_pos_weight` exactly as in the gradient.
        let scale_pos_weight = f64::from(self.scale_pos_weight);
        Some(Box::new(move |margin, label| {
            let weight = if label == 1.0 { scale_pos_weight } else { 1.0 };
            weight * 0.5 * (f64::from(margin) - f64::from(label)).powi(2)
        }))
    }

    fn default_metric(&self) -> EvalMetric {
        EvalMetric::Rmse
    }
}

/// Pseudo-Huber regression (`reg:pseudohubererror`): a smooth approximation of
/// the absolute error, robust to outliers. With residual `z = margin − y` and
/// slope `δ` (XGBoost `huber_slope`), the gradient is `z / √(1 + z²/δ²)` and
/// the Hessian `δ² / ((δ² + z²) · √(1 + z²/δ²))`, evaluated in `f32` exactly
/// like XGBoost's `PseudoHuberRegression`. The intercept is the trait's
/// default Newton step (XGBoost `FitIntercept`).
#[derive(Debug, Clone, Copy)]
pub(crate) struct PseudoHuberLoss {
    param: PseudoHuber,
    slope: f32,
}

impl PseudoHuberLoss {
    /// The loss with slope `param`.
    pub(crate) fn new(param: PseudoHuber) -> Self {
        PseudoHuberLoss {
            param,
            slope: param.slope() as f32,
        }
    }
}

impl Default for PseudoHuberLoss {
    fn default() -> Self {
        PseudoHuberLoss::new(PseudoHuber::default())
    }
}

impl Loss for PseudoHuberLoss {
    fn name(&self) -> &'static str {
        "reg:pseudohubererror"
    }

    fn gradient(
        &self,
        preds: &[f32],
        labels: &[f32],
        weights: Option<&[f32]>,
        out: &mut [GradPair],
    ) {
        let slope_sq = self.slope * self.slope;
        super::elementwise_gradient(preds, labels, weights, out, |p, y, w| {
            let z = p - y;
            let scale_sqrt = (1.0 + z * z / slope_sq).sqrt();
            let scale = slope_sq + z * z;
            GradPair::new((z / scale_sqrt) * w, (slope_sq / (scale * scale_sqrt)) * w)
        });
    }

    fn pointwise_loss(&self) -> Option<super::PointwiseLoss<'_>> {
        // `δ² (√(1 + z²/δ²) − 1)`, whose derivatives are the gradient above,
        // rationalized to `z² / (√(1 + z²/δ²) + 1)`: the subtraction would
        // cancel to `0` once `z²/δ²` drops below the `f64` epsilon (e.g. a
        // unit residual under a large slope), losing the quadratic regime.
        let slope_sq = f64::from(self.slope).powi(2);
        Some(Box::new(move |margin, label| {
            let z = f64::from(margin) - f64::from(label);
            z * z / ((1.0 + z * z / slope_sq).sqrt() + 1.0)
        }))
    }

    fn default_metric(&self) -> EvalMetric {
        EvalMetric::Mphe(self.param)
    }
}

/// Squared log error regression (`reg:squaredlogerror`): the loss
/// `½ [ln(1 + pred) − ln(1 + y)]²` behind the `rmsle` metric. Labels must
/// exceed `-1`. With the margin clamped to `p = max(margin, −1 + 10⁻⁶)` for
/// the gradient only, `g = [ln1p(p) − ln1p(y)] / (p + 1)` and
/// `h = max([−ln1p(p) + ln1p(y) + 1] / (p + 1)², 10⁻⁶)`, evaluated exactly
/// like XGBoost's `SquaredLogError` (the Hessian's square and division in
/// `f64`, as `std::pow(float, int)` promotes). The prediction transform is
/// the identity (predictions are not clamped) and the intercept is the
/// trait's default Newton step (XGBoost `FitIntercept`).
#[derive(Debug, Clone, Copy, Default)]
#[non_exhaustive]
pub struct SquaredLogError;

/// XGBoost's `fmaxf(predt, -1 + 1e-6)` bound: the `f64` constant rounded to
/// `f32`.
const SQUARED_LOG_MIN_PRED: f32 = (-1.0f64 + 1e-6) as f32;

impl Loss for SquaredLogError {
    fn name(&self) -> &'static str {
        "reg:squaredlogerror"
    }

    fn gradient(
        &self,
        preds: &[f32],
        labels: &[f32],
        weights: Option<&[f32]>,
        out: &mut [GradPair],
    ) {
        super::elementwise_gradient(preds, labels, weights, out, |p, y, w| {
            let p = p.max(SQUARED_LOG_MIN_PRED);
            let (log_p, log_y) = (p.ln_1p(), y.ln_1p());
            let grad = (log_p - log_y) / (p + 1.0);
            let shifted = f64::from(p + 1.0);
            let hess = ((f64::from(-log_p + log_y + 1.0) / (shifted * shifted)) as f32).max(1e-6);
            GradPair::new(grad * w, hess * w)
        });
    }

    fn validate_info(&self, info: &MetaInfo) -> Result<()> {
        // XGBoost `SquaredLogError::CheckLabel`: `log1p(label)` must be defined.
        check_label_domain(info, |y| y <= -1.0)
    }

    fn default_metric(&self) -> EvalMetric {
        EvalMetric::Rmsle
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::objective::{base_margins, gradient_pairs};

    #[test]
    fn gradient_matches_closed_form() {
        let obj = SquaredError::default();
        let preds = [2.0f32, 0.0, -1.0];
        let labels = [1.0f32, 0.5, -3.0];
        let out = gradient_pairs(&obj, &preds, &labels, None);
        assert_eq!(out[0], GradPair::new(1.0, 1.0)); // 2 - 1
        assert_eq!(out[1], GradPair::new(-0.5, 1.0)); // 0 - 0.5
        assert_eq!(out[2], GradPair::new(2.0, 1.0)); // -1 - (-3)
    }

    #[test]
    fn weighted_gradient_scales() {
        let obj = SquaredError::default();
        let preds = [2.0f32];
        let labels = [1.0f32];
        let w = [4.0f32];
        let out = gradient_pairs(&obj, &preds, &labels, Some(&w));
        assert_eq!(out[0], GradPair::new(4.0, 4.0));
    }

    /// `scale_pos_weight` multiplies the sample weight of the rows labeled
    /// exactly 1 only, and moves the intercept from the label mean to the
    /// Newton step on the reweighted rows: `Σ w'y / Σ w'`.
    #[test]
    fn scale_pos_weight_reweights_rows_labeled_one() {
        let obj = SquaredError::new(3.0);
        let out = gradient_pairs(&obj, &[2.0, 2.0], &[1.0, 1.5], Some(&[2.0, 2.0]));
        assert_eq!(out[0], GradPair::new(6.0, 6.0)); // (2 - 1) * 2 * 3
        assert_eq!(out[1], GradPair::new(1.0, 2.0)); // (2 - 1.5) * 2
        // (1 * 3 + 4) / (3 + 1) = 1.75, not the mean 2.5.
        assert_eq!(base_margins(&obj, &[1.0, 4.0], None), vec![1.75]);
    }

    #[test]
    fn base_margins_is_label_mean() {
        let obj = SquaredError::default();
        assert_eq!(base_margins(&obj, &[1.0, 2.0, 3.0], None), vec![2.0]);
    }

    /// Pseudo-Huber with slope δ: at `z = δ` the gradient is `δ/√2` and the
    /// Hessian `1/(2√2)`, so a wrong slope scaling would be visible.
    #[test]
    fn pseudo_huber_slope_scales_gradient() {
        let obj = PseudoHuberLoss::new(PseudoHuber::new(2.0).unwrap());
        let out = gradient_pairs(&obj, &[2.0], &[0.0], None);
        let root2 = 2f32.sqrt();
        assert!(
            (out[0].grad - 2.0 / root2).abs() < 1e-6,
            "grad {}",
            out[0].grad
        );
        assert!(
            (out[0].hess - 1.0 / (2.0 * root2)).abs() < 1e-6,
            "hess {}",
            out[0].hess
        );
        // Unit slope reproduces the classic form d/√(1+d²), 1/(1+d²)^{3/2}.
        let out = gradient_pairs(&PseudoHuberLoss::default(), &[2.0], &[0.0], None);
        let s = 5f32;
        assert_eq!(out[0], GradPair::new(2.0 / s.sqrt(), 1.0 / (s * s.sqrt())));
    }

    /// The pseudo-Huber intercept is the Newton step, not the label mean
    /// (XGBoost `FitIntercept`): for labels {0, 4} the mean is 2, but the
    /// bounded gradient `z/√(1+z²)` makes the step from zero, `-Σg/Σh` with
    /// `h = 1/(1+z²)^{3/2}`, fall well short of it.
    #[test]
    fn pseudo_huber_intercept_is_newton_step() {
        let obj = PseudoHuberLoss::default();
        let labels = [0.0f32, 4.0];
        let margins = base_margins(&obj, &labels, None);
        let s = 17f32; // 1 + 4²
        let g1 = -4.0f32 / s.sqrt();
        let h1 = 1.0f32 / (s * s.sqrt());
        let expected = (-f64::from(g1) / (1.0 + f64::from(h1))) as f32;
        assert_eq!(margins, vec![expected]);
        assert!(
            margins[0] < 1.0,
            "Newton step {} should undershoot the mean",
            margins[0]
        );
    }

    /// The loss keeps its quadratic regime `z²/2` under a slope so large that
    /// `1 + z²/δ²` rounds to `1`, and still matches `δ² (√(1 + z²/δ²) − 1)`
    /// where that form is accurate.
    #[test]
    fn pseudo_huber_loss_survives_large_slopes() {
        let obj = PseudoHuberLoss::new(PseudoHuber::new(1e9).unwrap());
        let loss = obj.pointwise_loss().unwrap();
        assert!((loss(0.0, 1.0) - 0.5).abs() < 1e-12, "{}", loss(0.0, 1.0));

        let obj = PseudoHuberLoss::new(PseudoHuber::new(2.0).unwrap());
        let loss = obj.pointwise_loss().unwrap();
        let naive = 4.0 * ((1.0f64 + 9.0 / 4.0).sqrt() - 1.0);
        assert!((loss(3.0, 0.0) - naive).abs() < 1e-12);
    }

    /// The squared-log gradient vanishes where `pred == label` and the
    /// margin clamp keeps it finite for margins at or below `-1`, where
    /// `ln1p` itself is undefined.
    #[test]
    fn squared_log_gradient_zero_at_label_and_finite_below_minus_one() {
        let obj = SquaredLogError;
        let out = gradient_pairs(&obj, &[3.0, -1.0, -7.0], &[3.0, 0.5, 0.5], None);
        assert_eq!(out[0].grad, 0.0);
        // Hessian at the optimum is 1 / (p + 1)².
        assert_eq!(out[0].hess, 1.0 / 16.0);
        assert!(out[1].grad.is_finite() && out[1].grad < 0.0);
        assert_eq!(out[1], out[2], "margins below the clamp share its gradient");
    }

    /// Far above the label the curvature term goes negative; XGBoost floors
    /// it at `1e-6` (times the weight).
    #[test]
    fn squared_log_hessian_floor_is_weighted() {
        let obj = SquaredLogError;
        let out = gradient_pairs(&obj, &[1e4], &[0.0], Some(&[2.0]));
        assert_eq!(out[0].hess, 2e-6);
        assert!(out[0].grad > 0.0);
    }

    #[test]
    fn squared_log_rejects_labels_at_minus_one() {
        let obj = SquaredLogError;
        assert!(
            obj.validate_info(&MetaInfo::new(&[-0.5, 2.0], None, None))
                .is_ok()
        );
        assert!(
            obj.validate_info(&MetaInfo::new(&[-1.0], None, None))
                .is_err()
        );
    }
}