Ultrafast artificial intelligence: machine learning with atomic-scale quantum systems

Abstract

We train a model atom to recognize pixel-drawn digits based on hand-written numbers in the range 0-9, employing intense light-matter interaction as a computational resource. For training, the images of the digits are converted into shaped laser pulses (data input pulses). Simultaneously with an input pulse, another shaped pulse (program pulse), polarized in the orthogonal direction, is applied to the atom and the system evolves quantum mechanically according to the time-dependent Schrödinger equation. The purpose of the optimal program pulse is to direct the system into specific atomic final states (classification states) that correspond to the input digits. A success rate of about 40% is achieved when using a basic optimization scheme that might be limited by the computational resources for finding the optimal program pulse in a high-dimensional search space. Our key result is the demonstration that the laser-programmed atom is able to generalize, i.e. successful classification is not limited to the training examples, but also the classification of previously unseen images is improved by training. This atom-sized machine-learning image-recognition scheme operates on time scales down to tens of femtoseconds, is scalable towards larger (e.g. molecular) systems, and is readily reprogrammable towards other learning/classification tasks. An experimental implementation of the scheme using ultrafast polarization pulse shaping and differential photoelectron detection is within reach.