Title: DIFFERENTIAL EQUATION NETWORKS

Abstract: Most deep neural networks use simple, fixed activation functions, such as sigmoids or rectified linear units, regardless of domain or network structure. We introduce differential equation networks, an improvement to modern neural networks in which each neuron learns the particular nonlinear activation function that it requires. We show that enabling each neuron with the ability to learn its own activation function results in a more compact network capable of achieving comperable, if not superior performance when compared to much larger networks. We also showcase the capability of a differential equation neuron to learn behaviors, such as oscillation, currently only obtainable by a large group of neurons. The ability of differential equation networks to essentially compress a large neural network, without loss of overall performance makes them suitable for on-device applications, where predictions must be computed locally. Our experimental evaluation of realworld and toy datasets show that differential equation networks outperform fixed activatoin networks in several areas.

1. INTRODUCTION AND RELATED WORK:

Driven in large part by advancements in storage, processing, and parallel computing, deep neural networks (DNNs) have become capable of outperforming other methods across a wide range of highly complex tasks. Although DNNs often produce better results than shallow methods from a performance perspective, one of the main drawbacks of DNNs in practice is computational expense. One could attribute much of the success of deep learning in recent years to cloud computing, and GPU processing. While deep learning based applications continue to be integrated into all aspects of modern life, future advancements will continue to be dependent on the ability to perform more operations, faster, and in parallel unless we make fundamental changes to the way these systems learn.

State-of-the-art DNNs for computer vision, speech recognition, and healthcare applications require too much memory, computation, and power to be run on current mobile or wearable devices. To run such applications on mobile, or other resource-constrained devices, either we need to use these devices as terminals and rely on cloud resources to do the heavy lifting, or we have to find a way to make DNNs more compact. For example, ProjectionNet by Ravi (Ravi, 2017) , and MobileNet by Howard et al. (Howard et al., 2017) are both examples of methods that use a compact DNN representations with the goal of on-device applications. In ProjectionNet, a compact "projection" network is trained in parallel to the primary network, and used for the on-device network tasks. MobileNet proposes a streamlined architecture in order to achieve network compactness. One drawback to these approaches is that network compactness is achieved at the expense of performance. In this paper, we propose a different method for learning compact, powerful, stand-alone networks; we allow each neuron to learn its individual activation function.

Up to this point, advancements in deep learning have been largely associated with one of four areas: optimization, regularization, activation, and network structure. Some of these advancements, such as network structure, address issues related to the type of problem at hand, and the nature of the data. Since the advent of the classic feed-forward neural network (FFNN), researchers have introduced highly effective network structures such as convolutional neural networks (CNN (Krizhevsky et al., 2012) , popular in computer vision, and Recurrent Neural Networks (RNN) for sequences. The advent of new activation functions such as rectified linear units (ReLU) (Nair & Hinton, 2010) , exponential linear units (ELU) (Clevert et al., 2015) , and scaled exponential linear units (SELU) (Klambauer et al., 2017) address a networks ability to effectively learn complicated functions, thereby allowing them to perform more complicated tasks. The choice of an activation function is typically determined empirically by tuning, or due to necessity. For example, in modern deep networks, the ReLU activation function is often favored over the sigmoid function, which used to be a popular choice in the earlier days of neural networks. A reason for this preference is that the ReLU function is nonsaturating and does not have the vanishing gradient problem when used in deep structures (Hochreiter, 1998) . Recently, searching algorithms have proven to be valuable for automatic construction of previously hand-designed network components and activation functions (Zoph et al., 2017; Bello et al., 2017; Ramachandran et al., 2017) . Zoph et al. (2017) ; Bello et al. (2017) ; Ramachandran et al. (2017) have proposed one such method for searching for more effective activation functions.

At first glance, and from an applicability perspective, our approach is similar to max-out networks (Goodfellow et al., 2013) , adaptive piecewise linear units by Agostinelli et al. (2014) , and Ramachandran et al. (2017) . However, in Ramachandran et al. (2017) , their functions do not transform themselves during network training, and all neurons utilize the same activation function. We introduce differential equation networks (DifENs), where the activation function of each neuron is the nonlinear, possibly periodic solution of a second order, linear, ordinary differential equation. While the number of parameters learned by max-out and PLU is proportional to the number of input weights to a neuron, and the number of linear units in that neuron, for each DifEN activation function we learn only five additional parameters.

One feature of DifENs that is particularly interesting within the healthcare industry is the ability of DifENs to regress decaying and periodic functions. Although efforts have been made to explore the usefulness of periodic functions in neural networks since the 1990s, their applicability has not yet been appreciated (Sopena et al., 1999) . Thanks to the successes of deep learning in recent years, researchers have begun re-exploring the potential of periodic functions as activations (Parascandolo et al., 2016) . This capability makes DifENs particularly well suited to problems where the desired output signals display combinations of periodicity and decay. Some examples include disease management, modeling chronic conditions such as exacerbation of COPD, medication adherence, and acute conditions that are expected to get better over time such as cancer with a successful course of treatment. Inversely, healthiness indices that might decay over time, and that could drastically effect a patients health if effectively modelled (e.g. cancer with an unsuccessful course of treatment).

Our contributions in this paper include the following: We introduce differential equation networks, which can learn complex concepts in compact representations. We propose a learning process to learn the parameters of a differential equation for each neuron. Finally, we empirically show that a differential equation network is effective when applied to real-world problems, and that they are capable of changing activation functions during the learning process.

2. NETWORK COMPRESSION:

The size of a neural network is delineated its number of hidden neurons and their interconnections, which together determine the network's complexity. The ideal size of a network depends on the intricacy of the concept it is required to learn. A network that is too small cannot entirely, and circumstantially learn a hard problem.

The existing, successful structures of neural networks Sutskever et al. (2014); LeCun et al. (2015) ; Bahdanau et al. (2014) use hundreds of thousands to millions of parameters to learn common tasks, such as translation or object recognition. Although these tasks are complex to model, and still challenging for modern neural networks (Sutskever et al., 2014) , they are considered "easy" when performed by a qualified human. It follows that, if we want to create a neural network with abilities that are beyond average human level aptitude, the size of such a network would need to grow considerably. For example, we have seen accuracy improvements in state-of-the-art DNNs applied to massive image classification problems resulting in part from a large increase in the number of layers He et al. (2016) . If this trend continues to subsist, shortly, we will witness neural networks with billions of parameters that are designed for sophisticated tasks such as live speech understanding and translation, as well as controlling robots or humanoids, which would require an immense increase in storage and computational resources.

For these reasons, we want to highlight the importance of efficient network design, and the opportunities presented by network contraction methodologies. We show that differential equation networks learn a given task with a smaller network when compared to a fixed activation DNNs.

3. METHOD:

Inspired by functional analysis and calculus of variations, instead of using a fixed activation function for each layer, we propose a novel solution for learning an activation function for each neuron in the network. In the results section we show that by allowing each neuron to learn its own activation function, the network as a whole can perform on par with (or even outperform) much larger baseline networks. In the supplementary material, we have briefly outlined how in Calculus of Variations, the Euler-Lagrange formula can be used to derive a differential equation from an optimization problem with function variables (Gelfand & Fomin, 1963; Gelfand et al., 2000) .

The idea of this paper is to find the parameters of an ordinary differential equation (ODE) for each neuron in the network, whose solution would be used as the activation function of the neuron. As a result, the differential equation network enables each neuron to learn a personalized activation function flexibly. We select (learn) the parameters of the differential equation from a low dimensional space (i.e., five). By minimizing the network loss function, our learning algorithm smoothly updates the parameters of the ODE, which would result in an uncountably1 extensive range of possible activation functions.

3.1. DIFFERENTIAL EQUATION ACTIVATION:

We parametrize the activation function of each neuron using a linear, second order ordinary differentiatial equation ay (t) + by (t) + cy(t) = g(t), parameterized by five coefficients (a ,b ,c ,c 1 ,c 2 ), which can be learned by the backpropagation algorithm. These coefficients are the only additional parameters that we learn for each neuron. a ,b and c are the scalars that we use to parametrize the ODE, and c 1 and c 2 represent the initial conditions of the ODE's solution. g(t) is a regulatory function that we have called the "core activation function". Mainly to simplify the math, and because it is a standard practice in control theory, we have set g(t) to the Heaviside step function. Therefore, the ODE that we have chosen is the following:

EQUATION

In engineering and physics, such a model is often used to denote the exchange of energy between mass and stiffness elements in a mechanical system, or between capacitors and inductors in an electrical system (Ogata & Yang, 2002) . Interestingly by using the solutions of this formulation as activation functions, we can gain a few key properties: approximation or reduction to some of the standard activation functions such as sigmoid or ReLU; the ability to capture oscillatory forms; and, exponential decay or growth. The latter commonly appear in health-care problems.

3.2. THE LEARNING ALGORITHM:

For fixed a, b and c, the solution of the differential equation will be

y = f (t; a, b, c)+c 1 f 1 (t; a, b, c)+ c 2 f 2 (t; a, b, c

) for some functions f , f 1 , f 2 . y lies on an affine space parametrized by scalars c 1 and c 2 that represent the initial conditions of the solution. As described in the following two subsections, our learning algorithm has two main parts: solving the differential equations once, and using a backpropagation-based algorithm for jointly learning the network weights and the five parameters of each neuron.

3.2.1. CLOSED-FORM SOLUTIONS:

First, we solve the differential equations parametrically and take the derivatives of the closed-form solutions: ∂y ∂t with respect to its input t, and ∂y ∂a , ∂y ∂b , ∂y ∂c with respect to parameter a, b, c. Moreover, the derivative with respect to c 1 and c 2 will be f 1 and f 2 , respectively. This is done once. We solved the equations and took their derivatives using the software package Maple (Maple 2018 ).

Maple also generates optimized code for the solutions, by breaking down equations in order to reuse computations.2 . Although we used Maple here, this task could have been done simply by pen and paper (although more time consuming).

Singularity of solutions. If one or two of the coefficients a, b or c are zero, then solution of the differential equation falls into a singularity subspace that is different than the affine function space of neighboring positive or negative values for those coefficients. For example, for b = 0 and a * c > 0 , the solution will be y(t) = sin

√ ct √ a c 2 + cos √ ct √ a c 1 -u(t) c cos √ ct √ a -1 , but for b = c = 0, we will have y(t) = 1/2 u(t)t 2 a + c 1 t + c 2 .

We observe that changing c > 0 to c = 0 will change the resulting activation function from a pure ocsillatory form to a (parametric) leaky rectified quadratic activation function. Our learning algorithm allows the activation functions to jump over the singularity subspaces. However, if they fall into a singular subspace, the derivative with respect the parameter that has become zero, and will be zero for the rest of training. Therefore, the training algorithm will continue to search for a better activation function only withing the singular subspace.

In practice, for some hyperparameter , if any one of a, b or c is less than , we project that value to exactly zero, and use the corresponding solution form the singular sub-space. We do not allow a = b = c = 0, in this rare case we force c = . During the learning process at most two of a, b and c can be zero, which creates seven possible subspaces (with a, b, c ∈ {R -{0}, {0}}) that are individually solved. Similarly, when b 2 -4ac is close to zero, the generic solution will be exponentially large, therefore if -< b 2 -4ac < , we explicitly set b = (4ac) to stablize the solution and to avoid large function values.

Approximation of Dirac's delta function. The derivative of activation function with respect to t will have Dirac's delta function δ(t) which is the derivative of the Heaviside function. In the parametric derivatives, we substituded the delta function with its approximation s * e -s * t /(1 + e -s * t )) 2 , which is the derivative of σ(s * t) = 1/(1 + e s * t )). This approximation is a commonly used in practice for the delta function (Zahedi & Tornberg, 2010 ). The larger s is, the more accurate the approximation of delta function will be.

In all of our experiments, we set = .01, and s = 100, although tuning might improve the results. 

3.2.2. GRADIENT DESCENT-BASED LEARNING:

We adopt regular backpropagation to update the values of parameters a, b, c, c 1 and c 2 for each neuron, along with using ∂y ∂t for updating network parameters w (input weights to the neuron), and for backpropagating the error to lower layers.

We initialize network parameters using current best practices with respect to the layer type (e.g. linear layer, convolutional layer, etc.). We initialize parameters a, b, c for all neurons with a random positive number less than one, and strictly greater than zero. We initialize c 1 = c 2 = 0.0. To learn the parameters θ = [a, b, c, c 1 , c 2 ] T along with the weights w on input values to each neuron, we deploy a gradient descent algorithm. Both the weights w, as well as θ are learned using the conventional backpropagation algorithm with Adam updates (Kingma & Ba, 2014) .

During training, we treat a, b, c, c 1 and c 2 like biases to the neuron (i.e., with input weight of 1.0) and update their values based on the direction of the corresponding gradients in each mini-batch.

3.3. DIFEN IS UNIVERSAL APPROXIMATOR:

Feedforward neural networks with monotonically-increasing activation functions are universal approximators (Hornik et al., 1989; Barron, 1993) . Networks with radial basis activation functions that are bounded, increasing and then decreasing are also shown to be universal approximators (Park & Sandberg, 1991) . In this subsection we show that DifEN is also a universal approximator. Lemma 1. If b 2 -4ac < 0, the solutions of ay (t) + by (t) + cy(t) = u(t) will oscillate with frequency

ω = √ 4ac-b 2 2a

, and in particular, if b = 0, then ω = c a .

Proof. If b 2 -4ac < 0, the roots of the characteristic equation of the ODE will be

-b ± i √ 4ac-b 2 2a

, where i = √ -1. By substituting the solutions in Euler's formula the resulting sine and cosine functions will have frequency ω = . In particular, if b = 0, we'll have ω = c a .

Moreover, the solution of the ODE ay (t) + cy(t) = u(t) for ac > 0 is y(t) = sin

√ ct √ a c 2 + cos √ ct √ a c 1 -u(t) c cos √ ct √ a

-1 . We will use this in the proof of the following theorem:

Theorem 2. Differential equation network is a universal approximator.

Proof. By Lemma 1, for b = 0 and c = 1, the oscilation frequency of the activation function of a neuron will be ω = 1/ √ a. Therefore, by varying a ∈ R + , a neuron activation function can generate all oscillation frequencies. Now, consider a neural network with one hidden layer of differential equation neurons and one linear output layer. For a target function h(t), we take the fourier transformation of h(t), and find the set of present frequencies Ω. For each present frequency ω i ∈ Ω, we add two differential equation neuron i 1 and i 2 to the hidden layer with c = 1, b = 0 and a = 1/ω 2 i . Let β i and γ i be the corresponding coefficient of the sine and cosine functions from the Fourier transformation. Then, we let c 1i1 = γ i , c 2i1 = β i and c 1i2 = c 2i2 = 0, and corresponding weights w i1 and w i2 from neurons i 1 and i 2 to the output linear layer to be equal to 1.0 and -1.0, respectively. This way, the two neurons will cancel the particular solution term of the function, and we'll have: y i1 (t) -y i2 (t) = β i sin(ω i t) + γ i cos(ωt). By construction, after adding all frequencies from the Fourier transformation of h(t), the neural network will output

i∈Ω β i sin(ω i t) + γ i cos(ωt) = h(t).

There are multiple other ways to prove this theorem. For example, since the sigmoid, ReLU, and ReLU-like functions are among the solutions of differential equations, we can directly apply results of the Stone-Weierstrass approximation theorem (De Branges, 1959) .

3.4. A GEOMETRIC INTERPRETATION:

The solution set of a differential equation forms a functional manifold that is affine with respect to c 1 and c 2 , but is nonlinear in a, b, and c. Clearly, this manifold has a trivially low dimensional representation in R 5 (i.e., {a, b, c, c 1 , c 2 }). Gradient descent changes the functionals in this low dimensional space, and the corresponding functional on the solution manifold is used as the learned activation function. Figure 2 attempts to visually explain how, for example, a ReLU activation function transforms to a cosine activation function. One DifEN neuron approximating the sine functions in Figure 3 shows an empirical example of this scenario where we initialized the activation functions with random parameters, and the learned differential equation coefficients {a, b, c} and initial condition coefficients {c 1 , c2} perfectly represented a sine function after training. In contrast, the learned model by ordinary fixed activation FFNNs were much less accurate with significantly larger networks. Figure 5 . in the appendix material visually shows how changing a parameter changes the activation function's behavior.

3.5. REDUCTION TO COMMON ACTIVATION FUNCTIONS:

Two common activation functions are the sigmoid σ(t) = 1 1+e -t , and the rectified linear unit ReLU(t) = max(0, t). The sigmoid function is a smooth approximation of the Heavyside step function, and ReLU can be approximated by integrating sigmoid of s * t for a large enough s: max(0, t) ≈ t -∞ 1 1+e -sz dz = log(1+e st )/s. Equivalently, y(t) = log(1+e t )+c 1 ≈ ReLU(t)+c 1 will be a solution of the following first order linear differential equation: y (t) = 1 1+e -st ≈ u(t) We can set g(t) to σ(t) = 1 1+e -st , or to the step function u(t). For g(t) = σ(t), the particular solutions of this differential equation when a = 0 involve the Gauss hypergeometric and Li 2 functions, which are expensive to evaluate. Fortunately, if we set the right hand side to u(t), then the particular solutions will only involve common functions such as linear, logarithmic, exponential, quadratic, trigonometric, and hyperbolic functions.

In practice, if the learning algorithm decides that a and b should be zero, we use g(t) = 1 1+e -st (i.e. y(t) = 1 c * (1+e -st ) ). Otherwise, we use a step function to avoid complex-valued solutions that involve special mathematical functions, and particularly for the reasons mentioned in the previous subsection. With these conditions in place, if a = 0, b = 0, and c = 1, we recover the sigmoid function; if a = 0, b = 1, and c = 0 we recover the ReLU function; if a = 1, b = 0, and c = 0 we obtain a parametric leaky rectified quadratic form y = ReLU(t) 2 + c 1 t + c 2 (similar to parametric leaky ReLU (He et al., 2015; Xu et al., 2015) ), which is the solution of y (t) = u(t). When b 2 -4ac < 0 we observe oscillatory behaviour. Depending on the sign of b, this can be decaying or exploding, but when b = 0 observe a purely oscilatory behavior.

The above-mentioned cases are only a few examples of solutions that could be chosen. The point to emphasize is that an extensive range of functions can be generated by simply varying these few parameters. (Figure 1 illistraits several examples.)

4. RESULTS AND DISCUSSION:

We have conducted several experiments to evaluate the performance and compactness of differential equation networks.

4.1. TOY REGRESSION DATASETS:

We ran two tests on toy regression problems. In these tests we compared fixed activation networks using ReLU, LeakyReLU and SELU activations to a significantly smaller DifEN.

Sine. We trained on two periods of a sine function and extrapolated half of one period of unseen values. As seen in Figure 3 , a single differental equation neuron can learn the function almost perfectly, while the fixed activation baseline networks fail to fit comperable to the training data, even with a 10x larger network.

A more sophisticated function. Next, we fit a more challnging function (y = (sin(t)-cos(2t) 2 )/2+ 4 * (1 + arccos(sin(t/2)))/3). A DifEN with 25 neurons in one hidden layer fits the function much more accurately than the fixed activation baseline networks with 250 neurons as shown in Figure 4 . 

4.2. CLASSIFICATION:

We compared the performance of DifEN with baselines on MNIST handwritten digit dataset. We compared the performance of fixed activation convolutional neural networks equipped with ReLU, SELU, or Swish (Ramachandran et al., 2017) activation functions to a smaller DifEN. The fixed activation networks all used the following architecture: Three convolutional layers consisting of 20 5x5 filters, 40 5x5 filters, and 60 4x4 filters respectively. A pooling layer and dropout layer was applied after the second and third activation layers, and two fully connected layers followed with 200 neurons in the first fully connected layer. The DifEN network comprised two convolutional layers with 20 5x5 filters and 40 3x3 filters, respectively. One pooling and one dropout layer were applied after the second activation. One fully connected layer was used at the end as opposed to two for the fixed activation networks, reducing the number of parameters further. Dropout probability, batch size, epochs and learning rate were consistent across all networks. The goal here was to test the assumption that a DifEN should be able to at least match the performance of a significantly larger, fixed activation network, thereby achieving network compression without sacrificing performance.

For these experiments, we ran three-fold cross-validation and reported the mean result for each network.

While the performance boost here is negligable, the notable achievement is that the CNN with DifEN activations was able to learn a representation on par with that of the significantly larger fixed activation networks.

4.3. REGRESSION:

We compare the performance of DifEN to that of fixed activation DNNs applied to a standard diabetes regression dataset. In this section we compared the performance of a simple, one hidden layer FFNN We ran the test in Figure 3 , numerous times with different initializations, each time the differential transformed itself to have a sine-like solution. This supports that DifEN activation functions transform during the training process. We also demonstrated the capability of DifENs to learning complex concepts, and with a significantly reduced network size. Table 2 shows a DifEN can perform on par, or better than a network with over 2x the number of parameters when compared to a fixed activation network. Moreover, DifENs can learn a better approximation when compared to a network with fixed activations throughout. The ability of DifENs to achieve top-notch performance with a compact representation makes them a good candidate for on-device applications. Moreover, DifENs seem well suited to applications that require the capabilities of a big neural network, but are currently limited by memory, or where latency issues are a factor, such as space applications and robotics.

5. CONCLUSION:

To the best of the authors' knowledge, the machine learning community has yet to explore differential equations in neural networks, dictionary learning (Mairal et al., 2009a; b; Zhang & Li, 2010) , or kernel methods (Shawe-Taylor & Cristianini, 2004; Schölkopf et al., 1999) . While the presented model, algorithms and results in this paper are the first application of ODEs in neural networks, they show a promising and successful example of the potential of differential equations in the development of new machine learning algorithm.

In this paper we introduced Differential Equation Networks (DifEN). We have showcased the ability of DifENs to learn complicated concepts with a compact network representation. We have demonstrated DifENs' potential to outperform conventional DNNs across a number of tasks, and with a reduced network size. Modern DNNs achieve performance gains in large by increasing the size of the network, which is not a sustainable trend. DifENs represent a new spin on deep learning, and possibly a way to expand the capabilities of machine learning and on-device AI. The proposed algorithm introduces a mechanism for learning the activation functions for each neuron in a network, and empirical results support a promising advancement in learning complex concepts in a compact representation. 10, b=0.50, c=0.50 a=0.42, b=0.50, c=0.50 a=0.73, b=0.50, c=0.50 a=1.05, b=0.50, c=0.50 a=1.37, b=0.50, c=0.50 a=1.68, b=0.50, c=0.50 a=2.00, b=0.50, c=0.50 (a) .50, b=0.10, c=0.50 a=0.50, b=0.42, c=0.50 a=0.50, b=0.73, c=0.50 a=0.50, b=1.05, c=0.50 a=0.50, b=1.37, c=0.50 a=0.50, b=1.68, c=0.50 a=0.50, b=2.00, c=0.50 (b) 50, b=0.50, c=0.10 a=0.50, b=0.50, c=0.42 a=0.50, b=0.50, c=0.73 a=0.50, b=0.50, c=1.05 a=0.50, b=0.50, c=1.37 a=0.50, b=0.50, c=1.68 a=0.50, b=0.50, c=2 .00 Figure 6 above shows that the differential equation network achieves performance on par with significantly larger fixed activation networks. We see that the fixed activation networks do not surpass the single neuron DifEN performance untill they 8 or more neurons.

:

Up to computational precision limitations.

The code will be accessible, upon request.

Figures and Tables:

Figure 1: A sample set of DifEN activation functions and their derivatives. The bold blue line is the activation fucntion, and the orange solid line is its derivative with respect to t. The dashed lines are its derivatives with respect to a, b and c. First and second on the top row from left are ReLU and ReQU. The bump in the derivative of ReLU is an artifact of approximating Dirac's delta.

Figure 2: Left: The solutions of ay + by + cy = u(t) lie on a manifold of functions. Right: Every point on this manifold can be equivalently represented as a point on a 5-dimentional space of {a, b, c, c 1 , c 2 } (three shown in the cartoon). The red arrow shows a path an initialized function takes to be gradually transformed to a different one.

Figure 3: Learning the sine function. The differential equation neuron learns the function significant;y better than larger baselines.

Figure 4: Fitting an arbitrary mathematical function.

Figure 5: The spectra of functions generated by varying one of a, b, c, and fixing the other two with c 1 = c 2 = 0.

Comparison of performance on MNIST by activation. Compactness DifEN performance is on par with significantly larger fixed activation networks

Table content: <table><tr><td/><td colspan="2">Activation ReLU</td><td colspan="3">LReLU SeLU</td><td>Swish</td><td>DifEN</td></tr><tr><td/><td>Accuracy</td><td colspan="3">0.9891 0.9890</td><td>0.9903 0.9918 0.9919</td></tr><tr><td colspan="6">equipped with fixed activations to that with DifEN neurons. Again, we use 3-fold cross validation and</td></tr><tr><td colspan="6">report the average performance. The following experiments were conducted using the open source</td></tr><tr><td colspan="2">diabetes regression data.</td><td/><td/><td/></tr><tr><td colspan="6">Table 2: Comparison of performance on diabetes regression dataset by network size and activation</td></tr><tr><td colspan="2">size DifEN</td><td>ReLU</td><td/><td colspan="2">LReLU</td><td>SeLU</td><td>Swish</td></tr><tr><td>1</td><td colspan="5">2490.781 7391.783 6977.1</td><td>6289.9</td><td>4298.493</td></tr><tr><td>2</td><td colspan="5">2446.003 3759.527 4562.308 3793.336 3249.608</td></tr><tr><td>4</td><td colspan="5">2412.504 2931.891 2720.555 2912.025 2839.323</td></tr><tr><td>8</td><td>2313.98</td><td colspan="2">2465.16</td><td colspan="2">2398.2</td><td>2488.361 2664.854</td></tr><tr><td>16</td><td>2117.47</td><td colspan="4">2334.454 2357.557 2165.465 2236.137</td></tr></table>

Peer review (iclr2019):
 Personal Expertise: The reviewer has extensive practical and theoretical experience with deep networks, different activation functions, as well as some practical experience using deep networks to model physical phenomena which are different inverse problems than the more common perception modeling. However, the reviewer is not knowledgeable in using ODEs for deep networks. Contributions of the paper : The contribution of the paper is in proposing a learnable activation function in form of an ODE which can help to better model highly oscillatory and irregular functions more efficiently. This can be potentially useful for special applications in inverse problems where (by field knowledge) we know highly non-linear and specific activation functions can be more reasonable than the common ReLU and its recent variants. Quality and composition: The composition of the theoretical part of the paper is clear. The experimental part is very limited though and does not include all the necessary details. Many of the details in the initial sections can be taken to the appendices to make room for more empirical studies. Novelty, related works: The work seems novel in proposing the specific activation function but in general there are many other works that propose learnable or more elaborate activation functions, neurons, or local parts of networks. It is not so clear how this work is different from those and nor is compared to those works. This includes networks in networks, maxout networks, capsule networks, etc. Critique of the theories and experiments: Theoretical Design: The theoretical design and derivation of the paper seem correct, although the reviewer is not an expert on this topic. However, it does not clearly mention why the ODE is not designed and solved for each problem separately. Should there be different design choices for y for each task/dataset? Why not solving the ODE during the training as well? If we are solving the ODE only once and based on some initialization of coefficients, it seems to be equivalent to designing a learnable activation function such as leaky ReLU. In that regard, one could call leaky-ReLU a DifEN? Experimental Setup: The motivation of the new activation function is for specific use-cases where oscillatory or decaying functions are to be modelled. In that respect, the experimental setup is quite limited and inconclusive. - MNIST experiment: to conclusively evaluate the performance of the proposed activation function, it is important to try fixed activation functions on the same architecture as the DifEN and vice versa. - Diabetes regression experiment: since the task is not a well-studied regression task, more experiments on various datasets are required to make a conclusion. - The learnable activation function can potentially make the network more prone to overfitting, this needs to be tested thoroughly. - An important application of the proposed activation function is mentioned to be for model compression. That should be properly tested on problems with large sets of parameters (such as ImageNet networks) and observe if the performance drop due to a decrease in the number of parameters in a standard network is sharper than that of a network with DifEN activations. - More tasks and more analysis should be performed for the real-world tasks that the authors mention as the motivation of this work (specifically medical diagnosis or predictions). The analysis should demonstrate and give insight on the extra generalization power that the new activation function brings to those problems. - Following on the previous point, it is important to empirically demonstrate for which applications DifEN is useful. - Some analyses are missing on when a neuron can accentuate the problem of vanishing and/or exploding gradients in certain configurations of the DifEN parameters. It seems like the situation can arise during the training where the activation function become too steep or too saturated. Summary judgment: All in all, I think the proposed work has some potential in specific applications, however, the experimental setup does not give a clear and conclusive message of where and how the new activation functions are useful.