P HYSICS - AWARE S PATIOTEMPORAL M ODULES WITH A UXILIARY T ASKS FOR M ETA -L EARNING
Anonymous authors
Paper under double-blind review
A BSTRACT
Modeling the dynamics of real-world physical systems is critical for spatiotemporal prediction tasks, but challenging when data is limited.
The scarcity of realworld data and the difﬁculty in reproducing the data distribution hinder directly applying meta-learning techniques.
Although the knowledge of governing partial differential equations (PDE) of the data can be helpful for the fast adaptation to few observations, it is mostly infeasible to exactly ﬁnd the equation for observations in real-world physical systems.
In this work, we propose a framework, physics-aware meta-learning with auxiliary tasks whose spatial modules incorporate PDE-independent knowledge and temporal modules utilize the generalized features from the spatial modules to be adapted to the limited data, respectively.
The framework is inspired by a local conservation law expressed mathematically as a continuity equation and does not require the exact form of governing equation to model the spatiotemporal observations.
The proposed method mitigates the need for a large number of real-world tasks for meta-learning by leveraging spatial information in simulated data to meta-initialize the spatial modules.
We apply the proposed framework to both synthetic and real-world spatiotemporal prediction tasks and demonstrate its superior performance with limited observations.
I NTRODUCTION
Deep learning has recently shown promise to play a major role in devising new solutions to applications with natural phenomena, such as climate change (Manepalli et al., 2019; Drgona et al., 2019), ocean dynamics (Cosne et al., 2019), air quality (Soh et al., 2018; Du et al., 2018; Lin et al., 2018), and so on.
Deep learning techniques inherently require a large amount of data for effective representation learning, so their performance is signiﬁcantly degraded when there are only a limited number of observations.
However, in many tasks in physical systems in the real-world we only have access to a limited amount of data.
One example is air quality monitoring (Berman, 2017), in which the sensors are irregularly distributed over the space – many sensors are located in urban areas whereas there are much fewer sensors in vast rural areas.
Another example is extreme weather modeling and forecasting, i.e., temporally short events (e.g., tropical cyclones (Racah et al., 2017b)) without sufﬁcient observations over time.
Moreover, inevitable missing values from sensors (Cao et al., 2018;
Tang et al., 2019) further reduce the number of operating sensors and shorten the length of fullyobserved sequences.
Thus, achieving robust performance from a few spatiotemporal observations in physical systems remains an essential but challenging problem.
Learning on a limited amount of data from physical systems can be considered as a few shot learning.
While recently many meta-learning techniques (Schmidhuber, 1987; Andrychowicz et al., 2016;
Ravi & Larochelle, 2017; Santoro et al., 2016; Snell et al., 2017;
Finn et al., 2017) have been developed to address this few shot learning setting, there are still some challenges for the existing meta-learning methods to be applied in modeling natural phenomena.
First, it is not easy to ﬁnd a set of similar meta-tasks which provide shareable latent representations needed to understand targeted observations.
For instance, while image-related tasks (object detection (He et al., 2017) or visual-question-answering tasks (Andreas et al., 2016; Fukui et al., 2016)) can take advantage of an image-feature extractor pre-trained by a large set of images (Deng et al., 2009) and well-designed architecture (Simonyan & Zisserman, 2014;
He et al., 2016; Sandler et al., 2018), there is no such large data corpus that is widely applicable for understanding natural phenomena.
Second, unlike computer vision or natural language processing tasks where a common object (images or words) is clearly de- ﬁned, it is not straightforward to ﬁnd analogous objects in the spatiotemporal data.
Finally, exact equations behind natural phenomena are usually unknown, leading to the difﬁculty in reproducing the similar dataset via simulation.
For example, although there have been some works (de Bezenac et al., 2018; Lutter et al., 2019; Greydanus et al., 2019) improving data efﬁciency via explicitly incorporating PDEs as neural network layers when modeling spatiotemporal dynamics, it is hard to generalize for modeling different or unknown dynamics, which is ubiquitous in real-world scenario.
In this work, we propose physics-aware modules designed for meta-learning to tackle the few shot learning challenges in physical observations.
One of fundamental equations in physics describing the transport of physical quantity over space and time is a continuity equation:
[Equation]
where ρ is the amount of the target quantity ( u ) per unit volume, J is the ﬂux of the quantity, and σ is a source or sink, respectively.
This fundamental equation can be used to derive more speciﬁc transport equations such as the convection-diffusion equation, Navier-Stokes equations, and Boltzmann transport equation.
Thus, the continuity equation is the starting point to model spatiotemporal (conservative) observations which are accessible from sensors.
Based on the form of ρ and J with respect to a particular quantity u , Eq.
1 can be generalized as:
[Equation]
where the function F ( · ) describes how the target u is changed over time from its spatial derivatives.
Inspired by the form of Eq.
2, we propose two modules: spatial derivative modules (SDM) and time derivative modules (TDM).
Since the spatial derivatives such as ∇ , ∇· , and ∇ 2 are commonly used across different PDEs, the spatial modules are PDE-independent and they can be meta-initialized from synthetic data.
Then, the PDE-speciﬁc temporal module is trained to learn the unknown function F ( · ) from few observations in the real-world physical systems.
This approach can effectively leverage a large amount of simulated data to train the spatial modules as the modules are PDE-independent and thus mitigating the need for a large amount of real-world tasks to extract shareable features.
In addition, since the spatial modules are universally used in physics equations, the representations from the modules can be conveniently integrated with datadriven models for modeling natural phenomena.
Based on the modularized PDEs, we introduce a novel approach that marries physics knowledge in spatiotemporal prediction tasks with metalearning by providing shareable modules across spatiotemporal observations in the real-world.
Our contributions are summarized below:
• Modularized PDEs and auxiliary tasks:
Inspired by forms of PDEs in physics, we decompose PDEs into shareable (spatial) and adaptation (temporal) parts.
The shareable one is PDE-independent and speciﬁed by auxiliary tasks: supervision of spatial derivatives .
• Physics-aware meta-learning: We provide a framework for physcis-aware meta-learning, which consists of PDE-independent/-speciﬁc modules.
The framework is ﬂexible to be applied to the modeling of different or unknown dynamics.
• Synthetic data for shareable modules: We extract shareable parameters in the spatial modules from synthetic data, which can be generated from different dynamics easily.
M ODULARIZED PDE S AND M ETA -L EARNING
In this section, we describe how the physics equations for conserved quantities are decomposable into two parts and how the meta-learning approach tackles the task by utilizing synthetic data when the data are limited.
2. D ECOMPOSABILITY OF V ARIANTS OF A C ONTINUITY E QUATION
In physics, a continuity equation (Eq. 1) describes how a locally conserved quantity such as temperature, ﬂuid density, heat, and energy is transported across space and time.
This equation underlies many speciﬁc equations such as the convection-diffusion equation and Navier-Stokes equations:
[Equation]
Navier-
[Equation]
eqn
[Equation]
) where the scalar u and vector ﬁeld u are the variables of interest (e.g., temperature, ﬂow velocity, etc.).
A dot over a variable is time derivative.
The common feature in these equations is that the forms of equations can be digested as (Bar-Sinai et al., 2019; Zhuang et al., 2020):
[Equation]
where the right-hand side denotes a function of spatial derivatives.
As the time derivative can be seen as a Euler discretization (Chen et al., 2018), it is notable that the next state is a function of the current state and spatial derivatives.
Thus, knowing spatial derivatives at time t is a key step for spatiotemporal prediction at time t + 1 for locally conserved quantities.
According to Eq. 3, the spatial derivatives are universally used in variants of Eq.
1 and only the updating function F ( · ) is speciﬁcally deﬁned for a particular equation.
This property implies that PDEs for physical quantities can be decomposable into two modules: spatial and temporal derivative modules.
2. S PATIAL D ERIVATIVE M ODULES : PDE- INDEPENDENT M ODULES
Finite difference method (FDM) is widely used to discretize a d -order derivative as a linear combination of function values on a n -point stencil.
[Equation]
where n > d .
According to FDM, it is independent for a form of PDE to compute spatial derivatives, which are input components of F ( · ) in Eq.
Thus, we can modularize spatial derivatives as
PDE-independent modules.
The modules that can be learnable as a data-driven manner to infer the coefﬁcients ( α i ) have been proposed recently (Bar-Sinai et al., 2019; Seo et al., 2020).
The datadriven coefﬁcients are particularly useful when the discretization in the n -point stencil is irregular and low-resolution where the ﬁxed coefﬁcients cause substantial numerical errors.
2. T IME D ERIVATIVE M ODULE : PDE- SPECIFIC M ODULE
Once upto d -order derivatives are modularized by learnable parameters, the approximated spatial derivatives from the spatial modules are fed into an additional module to learn the function F ( · ) in
Eq.
This module is PDE-speciﬁc as the function F describes how the spatiotemporal observations change.
Since the exact form of a ground truth PDE is not given, the time derivative module is datadriven and will be adapted to observations instead.
2. M ETA -L EARNING WITH PDE- INDEPENDENT /- SPECIFIC M ODULES
Recently, Raghu et al.
(2019) investigate the effectiveness of model agnostic meta-learning (MAML, Finn et al.
and it is found that the outer loop of MAML is more likely to learn parameters for reusable features rather than rapid adaptation.
The ﬁnding that feature reuse is the predominant reason for efﬁcient learning of MAML allows us to use additional information which is beneﬁcial for learning better representations.
Previously, the objective in meta-training has been considered to be matched with one in meta-test as the purpose of meta-learning is to learn good initial parameters applicable across similar tasks (e.g., image classiﬁcation to image classiﬁcation).
We are now able to incorporate auxiliary tasks under a meta-learning setting to reinforce reusable features for a main task.
As described in Sec. 2.1, the spatial modules are reusable across different observations, and thus, we can meta-initialize the spatial modules ﬁrst with spatial derivatives provided by synthetic datasets.
Then, we can integrate the spatial modules with the task-speciﬁc temporal module during meta-test to help adaptation of TDM on few observations.
Since the spatial modules are trained by readily available synthetic datasets, a large number of similar tasks for meta-training is not required.
[Figure]
P HYSICS - AWARE M ETA -L EARNING WITH A UXILIARY T ASKS
In this section, we develop a physics-aware meta-learning framework for the modularized PDEs.
Fig.
1 describes the proposed framework and its computational process.
3. S PATIAL D ERIVATIVE M ODULE
Input : Graph signals u i and edge features e ij = x
j − x i on G where x i is a coordinate of node i .
Output : Spatial derivatives { ˆ u k,i | i ∈ V and k ∈ K } where K = {∇ x , ∇ y , ∇ 2 x , ∇ 2 y } .
Require : Spatial derivative modules { φ k | k ∈ K } 1: for k ∈ K do do
5: end for 6: end for
As we focus on the modeling and prediction of sensor-based observations, where the available data points are inherently on a spatially sparse irregular grid, we use graph networks for each module φ k to learn the ﬁnite difference coefﬁcients (Bar-Sinai et al., 2019).
Given a graph G =
( V , E ) where V = { 1 , . . .
, N } and E
= { ( i, j ) :
i, j ∈ V } , a node i denotes a physical location x i =
( x i , y i ) where a function value u i = u
( x i , y i ) is observed.
Then, the graph signals with positional relative displacement as edge features are fed into the spatial modules to approximate spatial derivatives by Alg.
The coefﬁcients ( a i , b ( i,j ) ) on each node i and edge ( i, j ) are output of φ and they are linearly combined with the function values
u i and u j .
K denotes a set of ﬁnite difference operators.
For example, if we set K = {∇ x , ∇ y , ∇ 2 x , ∇ 2 y } , we have 4 modules which approximate ﬁrst/second order of spatial derivatives in 2-dimension, respectively.
3. T IME D ERIVATIVE M ODULE
Input : Graph signals u and approximated spatial derivatives ˆ u k where k ∈ K on G .
Time interval ∆ t
Output : Prediction of signals at next time step ˆ u ( t )
Require :
Time derivative module 1: ˆ u t =
TDM ( { u i , ˆ u k,i | i ∈ V and k ∈ K } ) 2:
ˆ u ( t ) =
u ( t − 1) + ˆ u t − 1 · ∆ t
Once spatial derivatives are approximated, another learnable module is required to combine them for a target task.
The form of line 2 in Alg. 2 comes from
Eq. 3 and TDM is adapted to learn the unknown function F ( · ) in the equation.
As our target task is the regression of graph signals, we use a recurrent graph network for TDM.
3. M ETA -L EARNING WITH A UXILIARY O BJECTIVE
As discussed in Sec. 2.1, it is important to know spatial derivatives at time t to predict next signals at time t + 1 for locally conserved physical quantities, however, it is impractical to access the spatial derivatives in the sensor-based observations as they are highly discretized over space.
In this section, we propose a physics-aware meta-learning framework to meta-initialize a spatial module by leveraging synthetic dataset with auxiliary tasks to provide reusable features for the main tasks: prediction spatiotemporal observations in the real-world.
The meta-initialization with the auxiliary tasks from synthetic datasets is particularly important.
First, the spatial modules can be universal feature extractors for modeling observations following unknown physics-based PDEs.
Unlike other domains such as computer vision, it has been considered that there is no particular shareable architecture for learning spatiotemporal dynamics from physical systems.
We propose that the PDE-independent spatial modules can be applicable as feature extractors across different dynamics as long as the dynamics follow a local form of conservation laws.
Second, we can utilize synthetic data to meta-train the spatial modules as they are PDE-agnostic.
This property allows us to utilize a large amount of synthetic datasets which are readily generated by numerical methods regardless of the exact form of PDE for targeted observations.
Finally, we can provide a stronger inductive bias which is beneﬁcial for modeling real-world observations but not available in the observations explicitly.
Algorithm
Input : A set of meta-train task datasets D
= {D 1 , . . .
, D B } where D b =
( D trb , D teb ) .
D b = { ( u bi , e bij , y ( a 1 ,b ) i , . . .
, y ( a K ,b ) i ) : i ∈ V b , ( i, j
) ∈
E b } where y ( a k , · ) i is an k -th auxiliary task label for the i -th node, given node/edge feature u b and e b , respectively.
Learning rate α and β .
Output : Meta-initialized spatial modules Φ .
[Equation] [Equation]
[Equation] [Equation]
[Equation] [Equation]
while
Alg. 3 describes how the spatial modules are meta-initialized by MAML under the supervision of
K different spatial derivatives.
First, we generate values and spatial derivatives on a 2D regular grid from an analytical function.
Then, we sample a ﬁnite number of points from the regular grid to represent discretized nodes and build a graph from the sampled nodes.
Each graph signal and its discretization becomes input feature of a meta-train task and corresponding spatial derivatives are the auxiliary task labels.
Fig.
2 visualizes graph signals and spatial derivatives for meta-initialization.
Once the spatial modules are initialized throughout meta-training, we reuse the modules for metatest where the temporal module (the head of the network) are adapted on few observations from real-world sensors (Alg. 4).
Although the standard MAML updates the body of the network (the spatial modules) as well, we only adapt the head layer ( θ ) as like almost-no-inner-loop method in Raghu et al.
The task at test time is graph signal prediction and the temporal modules ( θ ) are adapted by a regression loss function
L = (cid:80) Tt =
1 || u ( t ) − ˆ u ( t )
|| 2 on length T sequence ( D trm ) and evaluated on held-out ( t > T ) sequence ( D tem ) with the adapted parameters.
S PATIAL D ERIVATIVE M ODULES : R EUSABLE M ODULES
We have claimed that the spatial modules provide reusable features associated with spatial derivatives such as ∇ x u, ∇ y u , and ∇ 2 x u across different dynamics or PDEs.
While it has been shown that the data-driven approximation of spatial derivatives is more precise than that of ﬁnite difference method (Seo et al., 2020; Bar-Sinai et al., 2019), it is not guaranteed that the modules effectively provide transferrable parameters for different spatial resolution, discretization, and ﬂuctuation of function values.
We explore whether the proposed spatial derivative modules based on graph networks can be used as a feature provider for different spatial functions and discretization.
[Table]
We perform two sets of experiments: evaluate few-shot learning performance (1) when SDM is trained from scratch; (2) when SDM is metainitialized.
Fig.
2 shows how the graph signal and its discretization is changed over the different settings.
If the number of nodes is large, it can provide spatially high-resolution and thus, the spatial derivatives can be more precisely approximated.
Table 1 shows the parameters we used to generate synthetic datasets.
Note that metatest data is designed to evaluate interpolation/extrapolation properties.
Initial frequency decides the degree of ﬂuctuation (In Fig.
2, the middle one has higher F than that of the left one.).
For each parameter combination, we generate 100 different snapshots from the following form in Long et al.
[Equation]
where the index i denotes the i -th node whose coordinate is ( x i , y i ) in the 2D space ([0 , 2 π ] × [0 , 2 π ]) and k, l are randomly sampled integers.
From the synthetic data, the ﬁrst- and second-order derivatives are analytically given and SDM is trained to approximate them.
The prediction results for spatial derivatives are shown in Table 2.
The results show that the proposed module (SDM) is efﬁciently adaptable to different conﬁguration on few samples from metainitialized parameters compared to learning from scratch.
The ﬁnding implies that the parameters for spatial derivatives can be generally applicable across different spatial resolution, discretization, and function ﬂuctuation.
E XPERIMENTAL E VALUATION 5. P RELIMINARY : W HICH SYNTHETIC DYNAMICS NEED TO BE GENERATED ?
While Table 2 demonstrates that the PDE-independent representations are reusable across different conﬁgurations, it is still an open question: which topological conﬁguration needs to be used to construct the synthetic dynamics?
According to Table 2, the most important factor affecting error is
[Table]
[Figure]
an initial frequency ( F ), which determines min/max scales and ﬂuctuation of function values, and it implies that the synthetic dynamics should be similarly scaled to a target dynamics.
We use the same topological conﬁguration in Table 1 to generate synthetic datasets for a task in Section 5.2 and adapted conﬁguration for a task in Section 5.3.
We describe more details in Appendix B.
5. M ULTI - STEP G RAPH S IGNAL G ENERATION
Task : We adopt a set of multi-step spatiotemporal sequence generation tasks to evaluate our proposed framework.
In each task, the data is a sequence of L frames, where each frame is a set of observations on N nodes in space.
Then, we train an auto-regressive model with the ﬁrst T frames ( T -shot) and generate the following L − T frames repeatedly from a given initial input ( T -th frame) to evaluate its performance
[Table] [Table]
Datasets :
For all experiments, we generate meta-train tasks with the parameters described in
Table 1 and the target observations are 2 realworld datasets: (1) AQICO : national air quality index (AQI) observations (Berman, 2017); (2)
ExtremeWeather : the extreme weather dataset (Racah et al., 2017b).
For the AQI-CO dataset, we constructmeta-test tasks with the carbon monoxide (CO) ppm records from the ﬁrst week of each month in 2015 at land-based stations.
For the extreme weather dataset, we select the top-10 extreme weather events with the longest lasting time from the year 1984 and construct a meta-test task from each event with the
[Table]
observed surface temperatures at randomly sampled locations.
Since each event lasts fewer thanframes, each task has a very limited amount of available data.
In both datasets, graph signals are univariate.
Note that all quantities have ﬂuidic properties such as diffusive and convection.
Fig.shows the spatiotemporal dynamics of the extreme weather observations and sampled points.
More details are in the supplementary material.
Baselines : We evaluate the performance of a physics-aware architecture ( PA-DGN )
(Seo et al., 2020), which also consists of spatial derivative modules and recurrent graph networks (RGN), to see how the additional spatial information affects prediction performance for same architecture.
Note that PA-DGN has same modules in PiMetaL and the difference is that PiMetaL utilizes metainitialized spatial modules and PA-DGN is randomly initialized for learning from scratch on metatest tasks.
Additionally, the spatial modules in PA-DGN is replaced by ﬁnite difference method ( FDM+RGN ) to see if the numerical method provides better PDE-agnostic representations.
The baselines and PiMetaL are trained on the meta-test support set only to demonstrate how the additional spatial information is beneﬁcial for few-shot learning tasks.
Discussion : Table 3 shows the multi-step prediction performance of our proposed framework against the baselines on real-world datasets.
Overall, PA-DGN and PiMetaL show similar trend such that the prediction error is decreased as longer series are available for few-shot adaptation.
There are two important ﬁndings: ﬁrst, with the similar expressive power in terms of the number of learnable parameters, the meta-initialized spatial modules provide high quality representations which are easily adaptable across different spatiotemporal dynamics in the real-world.
This performance gap demonstrates that we can get a stronger inductive bias from synthetic datasets without knowing
PDE-speciﬁc information.
Second, the contribution of the meta-initialization is more signiﬁcant when the length of available sequence is shorter ( T = 5 ) and this demonstrates when the metainitialization is particularly effective.
Finally, the ﬁnite difference method provides proxies of exact spatial derivatives and the representations are useful particularly when T = 5 but its performance is rapidly saturated and it comes from the gap between the learnable spatial modules and ﬁxed numerical coefﬁcients.
The results provide a new point of view on how to utilize synthetic or simulated datasets to handle challenges caused by limited datasets.
5. G RAPH S IGNAL R EGRESSION
Task, datasets, and baselines : Defferrard et al.
(2019) conducted a graph signal regression task: predict the temperature x t from the temperature on the previous 5 days ( x t − 5 : x t − 1 ).
We split the GHCN dataset 1 spatially into two regions: (1) the USA (1,705 stations) and (2) Europe (EU) (703 stations) where there are many weather stations full functioning.
In this task, the number of shots is deﬁned as the number of input and output pairs to train a model.
As the input length is ﬁxed, more variants of graph neural networks are considered as baselines.
We concatenate the 5step signals and feed it into Graph convolutional networks ( GCN ) (Kipf & Welling, 2017), Graph attention networks ( GAT ) (Veliˇckovi´c et al., 2018), GraphSAGE (Hamilton et al., 2017), and Graph networks ( GN )
(Battaglia et al., 2018) to predict next signals across all nodes.
Discussion : Table 4 shows the results of the graph signal regression task across different baselines and the proposed method.
There are two patterns in the results.
First, although in general we observe an improvement in performance for all methods when we move from the 5-shot setting to the 10-shot setting, PiMetaL’s performance yields the smallest error.
Second, for the EU dataset, while 5-shot seems enough to achieve stable performance, it demonstrates that the PDE-independent representations make the regression error converge to a lower level.
Overall, the experimental results prove that the learned spatial representations from simulated dynamics are beneﬁcial for learning on limited data.
R ELATED W ORK
Physics-informed learning Since physics-informed neural networks are introduced in Raissi et al.
(2019), which ﬁnd that a solution of a PDE can be discovered by neural networks, physical knowledge has been used as an inductive bias for deep neural networks.
Advection-diffusion equation is incorporated with deep neural networks for sea-surface temperature dynamics (de Bezenac et al., 2018).
Lutter et al.
; Greydanus et al.
(2019) show that Lagrangian/Hamiltonian mechanics can be imposed to learn the equations of motion of a mechanical system and Seo & Liu (2019) regularizes a graph neural network with a speciﬁc physics equation.
Rather than using explicitly given equations, physics-inspired inductive bias is also used for reasoning dynamics of discrete objects (Battaglia et al., 2016;
Chang et al., 2016) and continuous quantities (Seo et al., 2020).
Long et al.
(2017; 2019) propose a numeric-symbolic hybrid deep neural network designed to discover
PDEs from observed dynamic data.
While there are many physics-involved works, to the best of our knowledge, we are the ﬁrst to provide a framework to use the physics-inspired inductive bias under the meta-learning settings to tackle the limited data issue which is pretty common for real-world data such as extreme weather events (Racah et al., 2017b).
Meta-learning
The aim of meta-learning is to enable learning parameters which can be used for new tasks unknown at the time of learning, leading to agile models which adapt to a new task utilizing only a few samples (Schmidhuber, 1987; Naik & Mammone, 1992; Thrun & Pratt, 1998).
Based on how the knowledge from the related tasks is used, meta-learning methods have been classiﬁed as optimization-based (Andrychowicz et al., 2016; Ravi & Larochelle, 2017; Duan et al., 2017; Finn et al., 2017; Nichol et al., 2018; Antoniou et al., 2018; Rusu et al., 2018; Grant et al., 2018), modelbased (Santoro et al., 2016; Munkhdalai & Yu, 2017; Duan et al., 2017; Mishra et al., 2018), and metric-based (Koch et al., 2015;
Vinyals et al., 2016;
Snell et al., 2017).
Recently, another branch of meta-learning has been introduced to more focus on ﬁnding a set of reusable modules as components of a solution to a new task.
Alet et al.
(2018; 2019) provide a framework, structured modular meta-learning, where a ﬁnite number of modules are introduced as task-independent modules and an optimal structure combining the modules is found from a limited number of data.
Chen et al.
(2019) introduces techniques to automatically discover task-independent/dependent modules based on Bayesian shrinkage to ﬁnd more adaptable modules.
To our knowledge, none of the above works provide a solution to use meta-learning for modeling physics-related spatiotemporal dynamics where it is hard to generate enough tasks for meta-initialization.
C ONCLUSION
In this paper, we propose a framework for physics-aware meta-learning with auxiliary tasks.
By incorporating PDE-independent/-invariant knowledge (spatial derivatives) from simulated data, the framework provide reusable features to meta-test tasks with a limited amount of data.
Experiments show that auxiliary tasks and physics-aware meta-learning help construct reusable modules that improve the performance of spatiotemporal predictions in real-world tasks where data is limited.
Although introducing auxiliary tasks based on synthetic datasets improves the prediction performance, they need to be chosen and constructed manually and intuitively.
Designing and identifying the most useful auxiliary tasks and data will be the focus of our future work.
A T ASK 1: M ULTI - STEP G RAPH S IGNAL G ENERATION A. M ETA - TRAIN
Data : For all experiments, we generate the data for meta-train tasks from a sum of sinusoidal functions with different spatial frequencies (Eq. 6).
[Equation]
where ( x, y ) in the 2D space ([0 , 2 π ] × [0 , 2 π ]) and k, l are randomly sampled integers.
Once the spatially continuous function values are generated, we uniformly sample different number of locations from all grid points as observed nodes to simulate the case where the observations are irregularly distributed in space.
We then construct a k -Nearest Neighbor graph based on the Euclidean distance as the input of graph neural net- works.
The combination of parameters to generate the synthetic dataset is given in Table 5.
We constructsnapshots per a combination of the parameters ( N, E, F ) using a unique random seed.
75 snapshots per each combination are used for D tr and 25 snapshots are for D te .
[Table]
Tasks :
For each node, we have the ﬁrst and second order derivatives.
We meta-train the spatial derivative modules (Sec. 3.1) to predict the spatial derivatives by feeding node and edge features (function value at a node and relative displacement, respectively) as input.
A. M ETA - TEST A.2. S YNTHETIC
Data : We generate the synthetic meta-test data from Eq. 6 but set different parameters to simulate the realistic scenario where meta-train tasks and meta-test tasks do not share the same distribution.
Tasks :
We reuse the spatial modules in A.1 to evaluate how the meta-initialized parameters are easily adaptable to unseen graph signals with different spatial resolution, discretization, and the degree of function ﬂuctuation.
We use 15 snapshots for the adaptation in meta-test and 75 snapshots are used to evaluate the proposed model.
[Figure] [Figure]
AQI-CO (Berman, 2017):
There are multiple pollutants in the dataset and we choose carbon monoxide (CO) ppm as a target pollutant in this paper.
We select sensors located in between latitude (26, 33) and longitude (115,125) (East region of China).
In this region, we sample multiple multivariate time series whose length should be larger than 12 steps (12 hours) for multiple meta-tasks.
There are around 60 working sensors and the exact number of the working sensors is varying over different tasks.
Fig.
4 shows the locations of selected AQI sensors.
ExtremeWeather :
We select the data in the year 1984 from the extreme weather dataset in (Racah et al., 2017a).
The data is an array of shape (1460, 16, 768, 1152), containing 1460 frames (4 per day, 365 days in the year).
16 channels in each frame correspond to 16 spatiotemporal variables.
Each channel has a size of 768 × 1152 corresponding to one measurement per 25 square km on earth.
For each frame, the dataset provides fewer than or equal to 15 bounding boxes, each of which labels the region affected by an extreme weather event and one of the four types of the extreme weather: (1) tropical depression, (2) tropical cyclone, (3) extratropical cyclone, (4) atmospheric river.
In the single feature setting, we only utilize the channel of surface temperature (TS).
Tasks
[Table]
AQI-CO : We select the ﬁrst sequence of carbon monoxide (CO) ppm records from each month in the year 2015 at land-based stations, and set up the meta-test task on each sequence as the prediction of CO ppm.
We
construct a 6-NN graph based on the geodesic distances among stations.
ExtremeWeather :
First, we aggregate all bounding boxes into multiple sequences.
In each sequence, all bounding boxes (1) are in consecutive time steps, (2) are affected by the same type of extreme weather, and (3) have an intersection over union (IoU) ratio above 0.25 with the ﬁrst bounding box in the sequence.
Then we select the top-10 longest sequences.
For each sequence, we consider its ﬁrst bounding box A as the
region affected by an extreme weather event, and extend it to a new sequence of 20 frames by cropping and appending the same region A from successive frames.
For each region we uniformly sample 10% of available pixels as observed nodes to simulate irregularly spaced weather stations and build a 4-NN graph based on the Euclidean distance.
Fig.
3 visualizes the ﬁrst 5 frames of one extended sequence.
In the single feature experiment, we set up a meta-test task on each extended sequence as the prediction of the surface temperature (TS) on all observed nodes with the initial TS given only.
A. E XPERIMENTAL D ETAILS A.3. B ASELINES
PA-DGN (train from scratch) (Seo et al., 2020):
For each meta-test task, initialize one PA-DGN model randomly and train it on the single task.
The spatial derivative layer uses a message passing neural network (MPNN) with 2 GN blocks using 2-layer MLPs as update functions.
The forward network part uses a recurrent graph neural network with 2 recurrent GN blocks using 2-layer GRU cells as update functions.
We set its hidden dimension to 64, in which case PA-DGN has a similar number of parameters with RGN.
The PA-DGN model has 384,653 learnable parameters.
A.3. O URS
PiMetaL : Meta-train the spatial derivative modules (SDM) with our proposed Alg. 3 on the meta-train tasks generated in A.1.
Then for each meta-test task, initialize one time derivative module (TDM) randomly and output of SDM is fed into the TDM to train it on the single task.
The architecture for SDM and TDM are identical for the spatial derivative layer and the recurrent graph network in PA-DGN, respectively.
A.3. T RAINING S ETTINGS
Training hyperparameters : For all meta-train and meta-test tasks, we use the Adam optimizer with the learn- ing rate 1e-3.
In each training epoch, we sample 1 task from all available tasks.
Environments : All experiments are implemented with Python3.6 and PyTorch 1.3.0, and are conducted with NVIDIA GTX 1080 Ti GPUs.
Runtime :
The
[Table]
GHCN (USA) (lon,lat) Synthetic for (USA) (x,y)
[Figure]
[Figure] [Figure]
[Figure] [Figure]
[Figure] [Figure]
S
i g n a l
GHCN
[Figure]
EU) (x
[Figure]
y
[Figure]
B T ASK 2: G RAPH S IGNAL R EGRESSION B. M ETA - TRAIN
Data : For the graph signal regression task, we generate synthetic dynamics for meta-train tasks and the syn- thetic data is adapted to a target dataset.
Before setting the topological conﬁguration for the synthetic dynamics, we ﬁrst examine the target dataset to understand its topological properties.
Based on the number of stations and the scale of records, we tune the topological conﬁguration for the synthetic dataset.
We use ( N, F ) = (1700 , 2) for the USA records and ( N, F ) =
(700 , 1 . 5) for Europe records, respectively, and 100 different initial values are generated to deﬁne different tasks.
Fig.
5 visualizes how the regional stations are distributed and Fig.
6 demonstrates how the spatial distribution of synthetic nodes and scales of synthetic values are adapted to the corresponding target dynamics.
B. M ETA - TEST
Data : The GHCN-Daily summaries from land surface stations across the globe provide daily climate records from numerous sources.
As the records from 100,000 stations in 180 countries and territories, the distribution of the weather stations is spatially non-uniform.
We sample sensors from two different regions (1) the USA and (2) Europe and construct a graph structure from the regional stations based on k -NN algorithm ( k = 4 ) as described in Defferrard et al.
There are 1,705 and 703 fully functioning sensors in the USA and Europe, respectively.
We use 2010 year records and ﬁrst few daily records for few-shot training (5 and 10) and next 100/150 days for validation and test.
Note that the number of learnable parameters is signiﬁcantly reduced compared to those of the previous task to minimize overﬁtting as well as be comparable to other variants of graph neural networks.
[Table]
B. E XPERIMENTAL D ETAILS B.3. B ASELINES
Since the input length of the regression task is ﬁxed (length=5), we can consider many variants of graph neural networks for the task.
We concatenate the 5-step signals and feed it into Graph convolutional net-
[Table] [Table]
[Table] [Table]
C S ENSITIVITY A NALYSIS OF S YNTHETIC D YNAMICS
It is important to study how much the model’s performance is dependent on synthetic topology.
In this section, we conduct an ablation study to see whether different choices of the synthetic topology affects the performance signiﬁcantly or not.
According to Table 4, the regression error is fairly converged from a few samples for the Europe records.
Thus, we apply different synthetic topology for the data to see if the saturated regression error is signiﬁcantly changed.
For this ablation study, we reuse the synthetic dynamics adapted for the USA records and generate one more synthetic dynamics for spatially low-resolution cases.
Table 7 shows that the regression performance across different topology is stable regardless of the number of shots, however, it is signiﬁcantly degraded when we change the synthetic topology from the adapted one ( ( N, F ) =
(700 , 1 . 5) ) for meta-training.
When we increase the spatial resolution ( N = 700 → 1700 ), the meta-initialized spatial modules are adapted to learn spatial derivatives deﬁned on spatially higher resolution.
In such case, SDM likely assigns high weights to directly adjacent nodes as well as farther nodes (e.g., 3-hop nodes) as all neighbor nodes are strongly associated with exact spatial derivatives.
On the other hand, if SDM is meta-initialized from a lower resolution ( N = 700 → 128 ), further nodes are too much underestimated.
Thus, it is important to construct proper topology for transferring the PDE-independent representations from synthetic dynamics to target dynamics.