Enhancing predictive imaging biomarker discovery through treatment effect analysis