Regression

Regression predicts a continuous target. It is common for activity values, binding affinity, energies, coordinates, expression levels, and other scalar or vector quantities.

A standard squared-error objective is:

Mean absolute error is less sensitive to large residuals:

Define the residual:

MSE weights large residuals quadratically:

MAE weights residuals linearly:

So the loss choice encodes whether large errors should dominate the update.

If uncertainty is modeled with a Gaussian likelihood:

then the model predicts both mean and uncertainty.

The negative log-likelihood is:

where is a constant independent of the model. This loss penalizes inaccurate means and overconfident uncertainty estimates.

If variance is fixed, Gaussian NLL reduces to MSE up to constants and scale:

A Laplace likelihood gives an absolute-error shape:

so:

Target Semantics

Regression is not only “predict a number.” The unit and transformation define the task:

  • Raw value: is used directly.
  • Log-transformed value: changes error interpretation.
  • Standardized value: requires recording train-set statistics.
  • Censored value: the true value is only known to be above or below a threshold.

If the target is standardized with train-set statistics:

then predictions must be transformed back before reporting metrics in physical units:

Do not fit normalization statistics on validation or test labels.

Output Shape

OutputExampleMetric Caution
scalaraffinity, energy, activity valueunit and transform dominate interpretation
vectormulti-property predictionaveraging can hide one bad endpoint
coordinateposition, pose, structurealignment, symmetry, and atom mapping matter
distributionmean and variancecalibration must be checked
censored valuemeasurement above/below thresholdordinary MSE may be invalid

Evaluation Boundary

Regression metrics answer different questions:

Metric TypeQuestion
MAEtypical absolute error in target units
RMSElarge-error-sensitive fit quality
variance explained relative to a baseline
Pearson/Spearmanlinear or rank association
NLL / calibrationare predicted uncertainties meaningful?

For downstream screening, a regression model may be used as a ranker. In that case Ranking metrics may be more relevant than absolute error.

Checks

  • Is the target scale linear, log-transformed, standardized, or censored?
  • Are outliers measurement errors or meaningful rare cases?
  • Does the metric reflect downstream tolerance, such as absolute error or rank quality?
  • Are uncertainty and calibration required for decision-making?
  • Are train-set normalization statistics reused consistently at validation and test time?
  • Are predictions reported in normalized units or original units?
  • Is the model evaluated as regression, ranking, or calibrated uncertainty?
  • Does the split prevent near-duplicate targets, proteins, molecules, or assays from crossing train and test?