Fig. 5: Model architecture. | Nature Communications