Small Proxy model to predict loss for given sample