Given that there are variables x1, x2, x3, and y, I know that x1, x2 are positively correlated with y, while x3 is negatively correlated with y, but the specific quantitative relationships are unknown. Furthermore, I have a dataset (X, y) pertaining to these variables, which is of limited size. To train an accurate model using this limited data, I wish to incorporate the prior knowledge about the correlations between x1, x2, x3, and y into the model, so as to utilize automatic differentiation frameworks like JAX to train a reliable model with a small amount of data.
Is there any method to build such a model and train it?