Methods, systems, apparatus, and computer-readable media for data band selection using machine learning. In some implementations, image data comprising information for each of multiple wavelength bands is obtained. A multi-layer neural network is trained using the image data to perform one or more classification or regression tasks. A proper subset of the wavelength bands is selected based on parameters of a layer of the trained multi-layer neural network, where the parameters were determined through training of the multi-layer neural network using the image data. Output is provided indicating that the selected wavelength bands are selected for the one or more classification or regression tasks.