Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural...

56
Can a machine infer quantum dynamics? Training a Recurrent Neural Network to Predict Quantum Trajectories from Raw Observation. Emmanuel Flurin* S. Hacohen-Gourgy, L. Martin, I. Siddiqi Quantum Nanoelectronics Laboratory, UC Berkeley *Quantronics, CEA Saclay, France arXiv:1811.12420

Transcript of Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural...

Page 1: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Can a machine infer quantum dynamics? Training a Recurrent Neural Network to Predict Quantum Trajectories from Raw Observation.

Emmanuel Flurin* S. Hacohen-Gourgy, L. Martin, I. Siddiqi

Quantum Nanoelectronics Laboratory, UC Berkeley *Quantronics, CEA Saclay, France

arXiv:1811.12420

Page 2: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones
Page 3: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones
Page 4: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Heralding ReadoutPrep TomoQuantum Evolution

500 m

A Quantum Information Processing ExperimentintroThe Quantum Slot Machine

Page 5: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Preparation Evolution Measurement

Black Box Quantum Mechanics

Page 6: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Quantum mechanics gives

Preparation Evolution Measurement

Black Box Quantum Mechanics

Page 7: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Preparation Evolution Measurement

Black Box Quantum Mechanics

Page 8: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Preparation Evolution Measurement

Black Box Quantum Mechanics

Page 9: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Preparation Evolution+Continuous Monitoring

Measurement

continuousmeasurement

Quantum mechanics gives

Stochastic Quantum Evolution

Page 10: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Preparation Evolution+Continuous Monitoring

Measurement

continuousmeasurement

« POVM » measurement backaction +

coherent dynamics

Stochastic Quantum Evolution

Physical parameters have to be separately calibrated and fine-tunedL.S. Martin

Page 11: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Supervised Deep Learning learns

continuousmeasurement

If one have a large set of instances

no matter how complicated the problem is

• language translation • image & speech recognition

• medical diagnosis • LHC signal processing

intro Inferring Quantum Dynamics

Page 12: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Superconducting circuits provides instances per minutes

Deep neural network can learn

with no prior on quantum mechanics

if spans a complete set of observable, is equivalent to the wave-function

Heralding ReadoutPrep TomoEvolution

Inferring Quantum Dynamics

Page 13: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro

Heralding ReadoutPrep TomoEvolution

preparationsettings

continuousmeasurement record

quantumtrajectory

data

«translation»

Inferring Quantum Dynamics

Page 14: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Circuit Quantum Electro-Dynamics

energyenergy

«matter» «light»

Circuits

Page 15: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Circuit Quantum Electro-Dynamics

energyenergy

dispersive coupling

Circuits

Page 16: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Physical Implementation

1 cm

Circuits

� ⇠ 2⇡ ⇥ 0.3 MHz

Page 17: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Experiment

effective frequency

reflectedphase

measurementfrequency

Dispersive Measurement

Page 18: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Experiment Josephson Amplifier

average measurement outcome

Page 19: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

STRONG vs. WEAK MEASUREMENT

Weak

Trajectories Strong vs Weak Measurement

weak interaction

projectivemeasurement

average measurement outcome

« Partial-Measurement Backaction and Nonclassical Weak Values in a Superconducting Circuit » Groen et al., PRL 2013 (Delft)

POVM : ⌦̂Vt

Page 20: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

STRONG vs. WEAK MEASUREMENT

Weak

Trajectories Strong vs Weak Measurement

weak interaction

projectivemeasurement

weak interaction

integrationtime

average measurement outcome

POVM : ⌦̂Vt

Page 21: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

STRONG vs. WEAK MEASUREMENT

Strong

Weak

Trajectories Strong vs Weak Measurement

weak interaction

projectivemeasurement

integrationtime

average measurement outcome

weak interaction

POVM : ⌦̂Vt

Page 22: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Pro

babi

lity

What does the detector signal tell us?

Homodyne Voltage

Bloch sphere

Quantum TrajectoriesTrajectories

Bayesian inference

Generalized measurement operator

Page 23: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Experimental data:

K. Murch et al., Nature 502 211 (2013).

Quantum TrajectoriesTrajectories

Page 24: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Weak Measurement of a Driven QubitTrajectories

JPA

Page 25: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

coun

ts

coun

ts

stochastic quantum evolution

preparation measurement

JPA

non-QNDmeasurement

Weak Measurement of a Driven Qubit

Recurent Neural Network

32 neurones per layer5,000 weights parameters0.8 ms of GPU training per trace

Experiment1.5 millions repetitionsat a rate of 0.5 ms

Long Short Term Memory

Trajectories

Page 26: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Deep Neural Network

hidden statesinput state output state

input output

each layer is representedas a vector of neurones

weight matrixconnecting each layer

sigmoidactivation function

neuronesbiases

Neural Network are differentiable computers

Neural Nets

Page 27: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets

hidden statesinput state output state

input output

gradient descent optimization

loss

(backpropagation)data

Cross-entropy loss function

minimum when the distribution of matches with

Deep Neural Network

measurement bit in the basis b prediction in the basis by 2 {0, 1}

Page 28: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets

hidden statesinput state output state

input output

gradient descent optimization

loss

(backpropagation)data

updating all the weight matrices with gradient descent of the loss function

backward propagation

(the chain rule for differentiation translates in matrix multiplications)

Deep Neural Network

Page 29: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets Deep Neural Network

hidden statesinput state output state

input output

gradient descent optimization

loss

(backpropagation)tomography

Page 30: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets Neural Networks Architectures

Time series• text generation • speech recognition (Amazon Alexa…) • language translation (Google translate…)

Recurrent Neural NetworkConvolutional Neural Network

Spatially ordered data• Image recognition

space-like correlations time-like causal correlations

Page 31: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets Recurrent Neural Networks

preparation settings inputed in ~h0

Page 32: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets Recurrent Neural Networks

loss

data

gradient descentoptimization

Page 33: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Nets Recurrent Neural Networks: Example

[Karpathy, Andrej, Justin Johnson, and Li Fei-Fei. "Visualizing and understanding recurrent networks. » arXiv preprint arXiv:1506.02078 (2015).]

character-level language modelingtrained on « War and Peace »input: current letter

output: most probable next letter

The sole importance of the crossing of the Berezina lies in the factthat it plainly and indubitably proved the fallacy of all the plans for cutting off the enemy's retreat and the soundness of the only possible line of action- -the one Kutuzov and the general mass of the army demanded- -namely, simply to follow the enem « y »?

Page 34: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

input: current letter

output: most probable next letter

Neural Nets

[Karpathy, Andrej, Justin Johnson, and Li Fei-Fei. "Visualizing and understanding recurrent networks. » arXiv preprint arXiv:1506.02078 (2015).]

Long range dependencies in RNN

Recurrent Neural Networks: Example

Page 35: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

RNN Learning Stochastic Quantum Dynamics

Recurent Neural Network

32 neurones per layer5,000 weights parameters0.8 ms of GPU training per trace

Experiment1.5 millions repetitionsat a rate of 0.5 ms

Long Short Term Memory

Prediction

stochastic quantum evolution

preparation measurement

JPA

quantum trajectory

Training set 6 preparation settings 6 measurement settings 20 experiment durations

LSTM babysitting batchsize 1024 10 epochs learning rate 10^{-3}->10^{-6} dropout 0.3->0

P (y|Z)

Page 36: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Weights Update

Prediction

strengthening

weakening

stochastic quantum evolution

preparation measurement

Recurrent Neural Network

JPA

RNN Learning Stochastic Quantum Dynamics

Recurent Neural Network

32 neurones per layer5,000 weights parameters0.8 ms of GPU training per trace

Experiment1.5 millions repetitionsat a rate of 0.5 ms

Long Short Term Memory

Training set 6 preparation settings 6 measurement settings 20 experiment durations

Page 37: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Neural Weights Update

Prediction

strengthening

weakening

stochastic quantum evolution

preparation measurement

Recurrent Neural Network

JPA

RNN Learning Stochastic Quantum Dynamics

Recurent Neural Network

32 neurones per layer5,000 weights parameters0.8 ms of GPU training per trace

Experiment1.5 millions repetitionsat a rate of 0.5 ms

Long Short Term Memory

Training set 6 preparation settings 6 measurement settings 20 experiment durations

Page 38: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

0

measurementpreparation

continuousmeasurement

RNN Training

stochastic master equation

RNN

Page 39: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Learning Quantum MechanicsRNN

continuousmonitoring

prep

meas

probabilityprobability

Page 40: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

ResultsTraining ValidationRNN

[Observing single quantum trajectories of a superconducting quantum bit Murch, Weber, Macklin, Siddiqi - Nature (2013)]

counts

training validation

Predictions

Tom

ogra

phy master equation

RNN

0.85

Page 41: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

ResultsTraining ValidationRNN

[Observing single quantum trajectories of a superconducting quantum bit Murch, Weber, Macklin, Siddiqi - Nature (2013)]

counts

training validation

Predictions

Tom

ogra

phy

master equationRNN

0.85

Page 42: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Training ValidationRNN

Page 43: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Machine PredictionRNN

Page 44: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Parameter Estimation & SensingRNN

JPA prediction distribution

Page 45: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Parameter Estimation & SensingRNN

JPA Drift map

Dissipative evolution

Page 46: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Parameter Estimation & SensingRNN

Dephasing rate

Rabi frequency

JPA Drift map

Dissipative evolution

Page 47: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Parameter Estimation & SensingRNN

JPA Drift map Diffusion map

quantum efficiency

Dephasing rate

Rabi frequency

Measurement back-actionDissipative evolution

Page 48: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Prediction and retrodiction with bi-directional RNN

Prediction

Retrodiction

[Prediction and retrodiction for a continuously monitored superconducting qubit D. Tan, S. Weber, I. Siddiqi, K. Mølmer, K. W. Murch PRL (2015)]

bi-RNN

[The two-state vector formalism of quantum mechanics Y Aharonov, L Vaidman (2002)]

stochastic quantum evolution

preparation measurement

Page 49: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Prediction and retrodiction with bi-directional RNN

Prediction

Retrodiction

[Prediction and retrodiction for a continuously monitored superconducting qubit D. Tan, S. Weber, I. Siddiqi, K. Mølmer, K. W. Murch PRL (2015)]

bi-RNN

[The two-state vector formalism of quantum mechanics Y Aharonov, L Vaidman (2002)]

stochastic quantum evolution

preparation measurement

continuousmonitoring

probabilityforward prediction

backward prediction

probability

Page 50: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

Prediction and retrodiction with bi-directional RNN

Prediction

Retrodiction

[Prediction and retrodiction for a continuously monitored superconducting qubit D. Tan, S. Weber, I. Siddiqi, K. Mølmer, K. W. Murch PRL (2015)]

bi-RNN

[The two-state vector formalism of quantum mechanics Y Aharonov, L Vaidman (2002)]

stochastic quantum evolution

preparation measurement

continuousmonitoring

probabilityforward prediction

backward prediction

probability

Tom

ogra

phy

Backward Predictions

Page 51: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

State Tomography from Retrodictionbi-RNN

Retrodiction

maximum likelihood reconstruction from 5000 trajectories

[Quantum state tomography with noninstantaneous measurements, imperfections, and decoherence Six, Campagne-Ibarcq, Dotsenko, Sarlette, Huard, and Rouchon PRA (2016)]

0

meas.prep

prep meas.

continuousmonitoring

prediction

retrodictionestimation(95% confidence)

initial state

Page 52: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

conclusion

Preparation Evolution Measurement

• Model independent validation of quantum trajectories beyond Markov approximation

• Efficient extraction of physical parameters - Quantum efficiency - Quantum state tomography

Black Box Experiement

Page 53: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

conclusion

Black box quantum machines will require black box models.

Big Quantum Machines

Page 54: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

conclusion

Noise models are not precisely known a priori: - non-markovian noise - Correlated errors

Big Quantum Machines

« Neural Decoder for Topological Codes »Torlai et al., PRL 2018

« Machine-learning-assisted correction ofcorrelated qubit errors in a topological code » Baireuther et al., arXiv

« Reinforcement Learning with Neural Networks for Quantum Feedback » Fösel et al., PRX (2018)

« Solving the quantum many-body problem with artificial neural networks »Carlea, Troyer. Science (2017)

Quantum many-body problem

Efficient design and decoding of Quantum Error Correction

Neural Networks can help to identify spatial and temporal hidden correlation

« Machine learning phases of matters » Carrasquilla, Melko. Nature Physics (2017)

Page 55: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

�55

S. Hacohen -Gourgy

L.S. Martin

Thanks

continuousmonitoring

prep

meas

arXiv:1811.12420

Page 56: Training a Recurrent Neural Network to Predict Quantum … · 2019. 5. 13. · Recurrent Neural Network JPA RNN Learning Stochastic Quantum Dynamics Recurent Neural Network 32 neurones

intro Quantum Master Equation

Preparation Evolution+Continuous Monitoring

Measurement

continuousmeasurement

with stochastic master equation

and Each of these terms has to be fine-tuned and separately calibrated