Source: http://www.sumobrain.com/patents/wipo/Systems-methods-activation-functions-photonic/WO2019217836A1.html
Timestamp: 2019-12-11 00:17:54
Document Index: 210819065

Matched Legal Cases: ['Application No. 62', 'Application No. 62', 'art 610', 'art 610', 'art 610', 'art 620', 'art 705', 'art 705', 'art 710', 'art 705', 'art 710', 'art 715', 'art 715', 'art 715', 'art 715', 'arts 910']

SYSTEMS AND METHODS FOR ACTIVATION FUNCTIONS FOR PHOTONIC NEURAL NETWORKS - THE BOARD OF TRUSTEES OF THE LELAND STANFORD JUNIOR UNIVERSITY
SYSTEMS AND METHODS FOR ACTIVATION FUNCTIONS FOR PHOTONIC NEURAL NETWORKS
WIPO Patent Application WO/2019/217836
Systems and methods for activation in an optical circuit in accordance with embodiments of the invention are illustrated. One embodiment includes an optical activation circuit, wherein the circuit comprises a directional coupler, an optical-to-electrical conversion circuit, a time delay element, a nonlinear' signal conditioner, and a phase shifter. The directional coupler receives an optical input and provides a first portion to the optical-to-electrical conversion circuit and a second portion to the time delay element, the time delay element provides a delayed signal to the phase shifter, and the optical-to-electrical conversion circuit converts an optical signal from the directional coupler to an electrical signal used to activate the phase shifter to shift the phase of the delayed signal.
HUGHES, Tyler, William (Office Of The General Counsel, Building 170 3rd Floor,Main Quad, P.O. Box 2038, Stanford CA, 94305-2038, US)
MINKOV, Momchil (Office Of The General General Counsel Building 170, 3rd Floor Main Quad P.O. Box 2038, Stanford CA, 94305-2038, US)
WILLIAMSON, Ian (Office Of The General Counsel Building 170, 3rd Floor Main Quad P.O. Box 2038, Stanford CA, 94305-2038, US)
FAN, Shanhui (Office Of The General Counsel Building 170, 3rd Floor Main Quad P.O. Box 2038, Stanford CA, 94305-2038, US)
US2019/031748
THE BOARD OF TRUSTEES OF THE LELAND STANFORD JUNIOR UNIVERSITY (Building 170, 3rd Floor Main Quad,,P.O. Box 2038, Stanford CA, 94305-2038, US)
G02F1/21; G02F1/01; G02F1/017; H03K3/42; H04B10/2575; H04B10/516
FINE, Isaac, M. (KPPB LLP, 2190 S. Towne Centre Place Suite 30, Anaheim CA, 92806, US)
1. An optical activation circuit, wherein the circuit comprises:
an optical-to-electrical conversion circuit;
a time delay element;
a nonlinear signal conditioner; and
a phase shifter, wherein:
the directional coupler receives an optical input and provides a first portion to the optical-to-electrical conversion circuit and a second portion to the time delay element;
the time delay element provides a delayed signal to the phase shifter; and
the optical-to-electrical conversion circuit converts an optical signal from the directional coupler to an electrical signal used to activate the phase shifter to shift the phase of the delayed signal.
2. The circuit of claim 1, wherein the nonlinear signal conditioner performs a nonlinear transformation of a voltage from the optical-to-electrical conversion circuit.
3. The circuit of claim 2, further comprising an element to add a static bias voltage to the electrical signal used to activate the phase shifter.
4. The circuit of claim 1, wherein the phase shifter is embedded in an interferometer to modulate the intensity of the delayed signal.
5. The circuit of claim 4, wherein the interferometer is a Mach-Zehnder interferometer.
6. The circuit of claim 1, wherein the optical-to-electrical conversion circuit comprises a photodetector.
7. The circuit of claim 6, wherein the optical-to-electrical conversion circuit further comprises a signal amplifier.
8. The circuit of claim 7, wherein the signal amplifier is a semiconductor optical amplifier.
[0001] This invention was made with Government support under contract FA9550-17-1-0002 awarded by the Air Force Office of Scientific Research. The Government has certain rights in the invention.
[0002] The present application claims the benefit of and priority to U.S. Provisional Patent Application No. 62/669,899 entitled’Training of Photonic Neural Networks Through in Situ Back- propagation”, filed May 10, 2018 and to U.S. Provisional Patent Application No. 62/815,243 entitled’’Systems and Methods for Activation Functions for Photonic Neural Networks”, filed March 7, 2019. The disclosure of U.S. Provisional Patent Application Serial Nos. 62/669,899 and 62/815,243 are herein incorporated by reference in its entirety.
[0003] The present invention generally relates to photonic neural networks and more specifically relates to activation functions for forward and/or backward propagation through photonic neural networks.
[0004] Recently, integrated optics has gained interest as a hardware platform for implementing machine learning algorithms, including artificial neural networks (ANNs), which rely heavily on matrix- vector multiplications that may be done efficiently in photonic circuits. Artificial neural net- works, and machine learning in general, are becoming ubiquitous for an impressively large number of applications. This has brought ANNs into the focus of research in not only computer science, but also electrical engineering, with hardware specifically suited to perform neural network operations actively being developed. There are significant efforts in constructing artificial neural network ar- chitectures using various electronic solid-state platforms, but ever since the conception of ANNs, a hardware implementation using optical signals has also been considered. Optical hardware platforms are particularly appealing for computing and signal processing due to their ultra-large signal bandwidths, low latencies, and reconfigurability. Photonic implementations benefit from the fact that, due to the non-interacting nature of photons, linear operations - like the repeated matrix multiplications found in every neural network algorithm - can be performed in parallel, and at a lower energy cost, when using light as opposed to electrons.
[0005] Systems and methods for activation in an optical circuit in accordance with embodiments of the invention are illustrated. One embodiment includes an optical activation circuit, wherein the circuit comprises a directional coupler, an optical-to-electrical conversion circuit, a time delay element, a nonlinear signal conditioner, and a phase shifter. The directional coupler receives an optical input and provides a first portion to the optical-to-electrical conversion circuit and a second portion to the time delay element, the time delay element provides a delayed signal to the phase shifter, and the optical-to-electrical conversion circuit converts an optical signal from the directional coupler to an electrical signal used to activate the phase shifter to shift the phase of the delayed signal.
[0006] In a further embodiment, the nonlinear signal conditioner performs a nonlinear transformation of a voltage from the optical-to-electrical conversion circuit.
[0007] In still another embodiment, the method further includes steps for an element to add a static bias voltage to the electrical signal used to activate the phase shifter.
[0008] In a still further embodiment, the phase shifter is embedded in an interferometer to modulate the intensity of the delayed signal.
[0009] In yet another embodiment, the interferometer is a Mach-Zehnder interferometer.
[0010] In a yet further embodiment, the optical-to-electrical conversion circuit includes a photodetector.
[0011] In another additional embodiment, the optical-lo-electiical conversion circuit further in- eludes a signal amplifier.
[0012] In a further additional embodiment, the signal amplifier is a semiconductor optical amplifier.
[0013] Additional embodiments and features are set forth in part in the description that follows, and in part will become apparent to those skilled in the art upon examination of the specification or may be learned by the practice of the invention. A further understanding of the nature and advantages of the present invention may be realized by reference to the remaining portions of the specification and the drawings, which forms a part of this disclosure.
[0014] The description and claims will be more fully understood with reference to the follow- ing figures and data graphs, which are presented as exemplary embodiments of the invention and should not be construed as a complete recitation of the scope of the invention.
[0015] Figure 1 illustrates a block diagram of a feedforward neural network (or a photonic hardware platform) of L layers in accordance with an embodiment of the invention.
[0016] Figure 2 illustrates an example of a single layer of a feedforward neural network in ac- cordance with an embodiment of the invention.
[0017] Figure 3 illustrates an example of an activation block in accordance with an embodiment of the invention.
[0018] Figure 4 illustrates responses across a representative selection of electrical biasing in accordance with an embodiment of the invention.
[0019] Figure 5 illustrates a chart of the relationship between optical-to-electrical gain G and the modulator V n for various activation thresholds.
[0020] Figure 6 illustrates charts of a nonlinear parameter for the electro-optic activation as a function of gain and modulator.
[0021] Figure 7 illustrates charts of an XOR example.
[0022] Figure 8 illustrates charts of a handwritten number recognition example. [0023] Figure 9 illustrates charts of comparisons between the validation accuracy and the loss function value during training for a two-layer network.
[0024] Figure 10 illustrates a confusion matrix for the trained ONN with an electro-optic activation function.
[0025] Turning now to the drawings, systems and methods in accordance with certain embodi- ments of the invention can be used to train and implement photonic neural networks. Systems and methods in accordance with many embodiments of the invention provide an electro-optic hardware platform for nonlinear activation functions in optical neural networks. The optical-to-optical non- linearity in accordance with many embodiments of the invention operates by converting a small portion of the input optical signal into an analog electric signal, which can be used to intensity- modulate the original optical signal with no reduction in operating speed. In some embodiments, this scheme can allow for complete nonlinear on-off contrast in transmission at relatively low optical power thresholds and can eliminate the requirement of having additional optical sources between each layer of the network. In numerous embodiments, the activation function is recon- figurable via electrical bias, allowing it to be programmed or trained to synthesize a variety of nonlinear responses. Activation functions in accordance with various embodiments of the inven- tion can significantly improve the expressiveness of optical neural networks, allowing them to perform well on machine learning tasks. Although many of the examples described herein are de- scribed with reference to a particular hardware implementation of a photonic ANN, one skilled in the art will recognize that methods and systems can be readily applied to other photonic platforms without departing from the heart of the invention.
[0026] Nonlinear activation functions play a key role in ANN s by enabling them to learn complex mappings between their inputs and outputs. Whereas digital processors have Ihe expressiveness to trivially apply nonlinearities such as the widely-used sigmoid, ReLU, and tanh functions, the realization of nonlinearities in optical hardware platforms is more challenging. One reason for this is that optical nonlinearities are relatively weak, necessitating a combination of large inter- action lengths and high signal powers, which impose lower bounds on the physical footprint and the energy consumption, respectively. Although it is possible to resonantly enhance optical non- linearities, this comes with an unavoidable hade-off in reducing the operating bandwidth, thereby limiting the information processing capacity of an ONN. Additionally, maintaining uniform reso- nanl responses across many elements of an optical circuit necessitates additional control circuitry for carefully calibrating each element.
[0027] A more fundamental limitation of optical nonlinearities is that their responses tend to be fixed during device fabrication. This limited tunability of the nonlinear optical response prevents an ONN from being reprogrammed to realize different nonlinear activation functions, which may be more suitable for a given machine learning task. Similarly, a fixed nonlinear response may also limit the performance of very deep ONNs with many layers of activation functions when the optical signal power drops below the activation threshold due to loss in previous layers. The activation threshold corresponds to the point on activation transfer function where nonlinearity is strongest. For example, with optical saturable absorption from 2D materials in waveguides, the activation threshold is on the order of 1-10 mW, meaning that the strength in the nonlinearity of each subsequent layer will be successively weaker.
[0028] In light of these challenges, other methods have attempted to implement activation func- tions by detecting each optical signal, feeding them through a conventional digital computer to apply the nonlinearity, and then modulating new optical signals for the subsequent layer. Although such approaches benefit from the flexibility of digital signal processing, conventional processors have a limited number of input and output channels, which would prevent such approaches from scaling to very matrix dimensions and a large number of optical inputs. Moreover, digitally ap- plied nonlinearities add latency from the analog-to-digital conversion process and constrain the computational speed of the neural network to the same GHz-scale clock rates which ONNs seek to overcome. [0029] Systems and methods in accordance with some embodiments of the invention provide an electro-optic architecture for synthesizing optical-to-optical nonlinearities which alleviates the is- sues discussed above. In many embodiments, architectures can feature complete on-off contrast in signal transmission, a variety of nonlinear response curves, and a low activation threshold. Rather than using traditional optical nonlinearities, systems and methods in accordance with a variety of embodiments of the invention can operate by measuring a small portion of the incoming optical sig- nal power and using electro-optic modulators to modulate the original optical signal, without any reduction in operating bandwidth or computational speed. Additionally, processes in accordance with certain embodiments of the invention can allow for the possibility of performing additional nonlinear transformations on the signal using analog electrical components. Many of the exam- pies described herein focus on the application of the electro-optic architecture as an element-wise activation in a feedforward ONN, but one skilled in the art will recognize that the synthesis of low-threshold optical nonlinearlities could be of broader interest in a variety of applications and fields such as (but not limited to) optical computing and information processing.
[0030] An ANN (or feedforward neural network) is a function which accepts an input vector, xo and returns an output vector, Specifically, ANNs can perform several layers of transforma- tions on their inputs, with each consisting of a linear matrix- vector multiplication followed by the application of an element-wise nonlinear function, or activation, on the result. Optical hardware implementations of ANNs have been proposed in various forms over the past few decades. In various embodiments, optical hardware implementations of an ANN implement linear operations using an integrated optical circuit. In numerous embodiments, the information being processed by the network, x,·, can be encoded into the modal amplitudes of the waveguides feeding the de- vice. Matrix-vector multiplications in accordance with many embodiments of the invention can be accomplished using meshes of integrated optical interferometers. Training the network in ac- cordance with numerous embodiments of the invention requires finding the optimal settings for the integrated optical phase shifters controlling the inteferomelers, which may be found using an analytical model of the chip, or using in-situ backpropagation techniques. [0031] A block diagram of a feedforward neural network 100 (or a photonic hardware platform) of L layers in accordance with an embodiment of the invention is illustrated in Figure 1. This figure illustrates that each layer of neural network 100 consists of a Wi block 110 representing a linear matrix which multiplies vector inputs c,· ~ i. In some embodiments, each W block represents an optical interference unit (OIU) with a number of integrated phase shifters. In many embodiments integrated phase shifters can be used to control an OIU and train a network.
[0032] Each layer of neural network 100 in this example also includes an activation block 120 that represents an element-wise nonlinear activation function operating on vectors z,· to produce outputs jc,. For a layer with index i, containing a weight matrix W and activation function this operation can be described mathematically as
[0033] Before they are able to perform useful computations, ANNs must be trained to accom- plish a given machine learning task. The training process is typically accomplished by minimizing the prediction error of the ANN on a set of training examples, which come in the form of input and target output pairs. For a given ANN, a loss function (or cost function) is defined over the target output and output predicted by the network. During training, this loss function is minimized with respect to tunable degrees of freedom, namely the elements of the weight matrix Wi within each layer.
[0034] An example of a single layer of a feedforward neural network in accordance with an embodiment of the invention is illustrated in Figure 2. This figure shows a schematic of a layer 200 with a weight matrix block (W) 210 and an activation block (ft) 220. In this example, weight matrix block 210 performs linear operations using an optical interference unit (OIU). OIUs in accordance with certain embodiments of the invention are a mesh of controllable Mach-Zehnder interferometers (MZIs) integrated in a silicon photonic circuit. By tuning the phase shifters inte- grated in the MZIs, any unitary N x. N operation on the input can be implemented, which finds applications both in classical and quantum photonics. In photonic ANNs in accordance with some embodiments of the invention, OIUs can be used for each linear matrix-vector multiplication. In certain embodiments, nonlinear activations can be performed using an electronic circuit, which involves measuring the optical state before activation, performing the nonlinear activation function on an electronic circuit such as a digital computer, and preparing the resulting optical state to be injected to the next stage of the ANN.
[0035] In the description of this example, the OIU is described by a number, N, of single-mode waveguide input ports coupled to the same number of single-mode output ports through a linear and lossless device. In certain embodiments, the device may also be extended to operate on differ- ing numbers of inputs and outputs. OIUs in accordance with some embodiments of the invention implement directional propagation such that all power flows exclusively from the input ports to the output ports. In its most general form, devices implement the linear operation where X ln and Z ou t are the modal amplitudes at the input and output ports, respectively, and W, or the transfer matrix, is the off-diagonal block of the system’s full scattering matrix,
[0036] The diagonal blocks are zero because forward-only propagation is assumed, while the off-diagonal blocks are the transpose of each other because a reciprocal system is assumed. Z in a nd Xou t correspond to file input and output modal amplitudes, respectively, if the device were run in reverse, i.e., sending a signal in from the output ports.
Nonlinear Activation Function [0037] Systems and methods in accordance with some embodiments of the invention provide a nonlinear activation function architecture for optical neural networks. Nonlinear activation func- tions in accordance with numerous embodiments of the invention can implement an optical-to- optical nonlinearity by converting a small portion of the optical input power into an electrical volt- age. The remaining portion of the original optical signal can be phase modulated by this voltage as it passes through an interferometer. In numerous embodiments, the resulting nonlinear optical activation function, f(z), of the input signal amplitude, z, is a result of an interferometer intensity modulation response as well as the components in the electrical signal pathway.
[0038] A more detailed example of an activation block (/)·) is illustrated in Figure 3. Activa- tion block 300 includes a directional coupler 310, optical-to-electrical conversion circuit 320, a nonlinear signal conditioner 330, time delay 340, and an interferometer 350. This figure shows a proposed optical-to-optical activation function in accordance with an embodiment of the invention which achieves a nonlinear response by converting a small portion of the optical input, z into an electrical signal, and then intensity modulating the remaining portion of the original optical signal as it passes through an interferometer.
[0039] In Figure 3, thick black lines represents optical waveguides, while thinner blue lines rep- resent electrical signal pathways. The input signal z first enters a directional coupler 310 which routes a portion of the input optical power, a, to a photodetector of optical-to-electrical conversion circuit 320. The photodetector is the first element of an optical-to-electrical conversion circuit 320, which is a standard component of high-speed optical receivers for converting an optical intensity into a voltage. In some embodiments, the optical-to-electrical conversion circuit also amplifies the optical intensity, such as (but not limited to) via a semiconductor optical amplifier (SOA) before the photodiode. In many embodiments, a transimpedance amplifier can be used to convert an output photocurrent to a voltage that can actuate a phase shifter, but optical amplification could alleviate the requirement of very high gain within the electrical domain. In certain embodiments, the required gain can be further reduced by utilizing an electro-optic phase shifter with a low threshold voltagge to induce a phase shift in a phase modulator (V n ). In this example, a normalization of the optical signal is assumed, such that the total power in the input signal is given by The optical-to-electrical conversion process in accordance with certain embodiments of the invention consists of a photodetector producing an electrical current, , where is the photode
tector responsivity. A transimpedance amplifying stage, characterized by a gain G, converts this current into a voltage The output voltage of the optical-to-electrical conversion
circuit then passes through a nonlinear signal conditioner 330 with a transfer function, H(-). In certain embodiments, this component allows for the application of additional nonlinear functions to transform the voltage signal. Finally, the conditioned voltage signal, H(VG ) can be added to with a static bias voltage, V¾ to induce a phase shift of for the optical signal routed through the lower port of the directional coupler 310. The parameter n p represents the voltage required to induce a phase shift of p in the phase modulator. This phase shift is a nonlinear self-phase modulation because it depends on the input signal intensity.
[0040] In this example, an optical delay line 340 between the directional coupler and the Mach- Zehnder interferometer (MZI) 350 is used to match the signal propagation delays in the optical and electrical pathways. In various embodiments, optical delay lines can ensure that the nonlinear self-phase modulation is applied at the same time that the optical signal which generated it passes through the phase modulator. In this example, Ihe optical delay is , accounting
for the contribution from the group delay of the optical-to-electrical conversion stage (toe), the delay associated with the nonlinear signal conditioner (t P i), and the RC time constant of the phase modulator
[0041] In a number of embodiments, the nonlinear self-phase modulation achieved by the electric circuit can be converted into a nonlinear amplitude response by a MZI, which has a transmission depending on Df as Depending on the configuration of the bias, V¾, a larger input optical signal amplitude can cause either more or less power to be diverted away from the output port, resulting in a nonlinear self- intensity modulation. Combining the expression for the nonlinear self-phase modulation with the MZI transmission, the mathematical form of the activation function can be written explicitly as
[0042] In some of the descriptions below, no nonlinear signal conditioning is applied to the elec- trical signal pathway, i.e. H(VG ) = VG · However, even with this simplification the activation function still exhibits a highly nonlinear response. Saturating effects in the OE conversion stage, which can occur in either the photodetector or the amplifier, are also neglected. However, in prac- tice these saturating effects could be taken advantage of to modify the optical-to-optical transfer function in accordance with numerous embodiments of the invention.
[0043] With the above simplifications, a more compact expression for the activation function response is
This equation indicates that the amount of phase shift per unit input signal power can be increased via the gain and photodiode responsivity, or by converting a larger fraction of the optical power to the electrical domain. However, tapping out a larger fraction optical power also results in a larger linear loss, which provides no benefit to the nonlinearity. [0044] In many embodiments, the electrical biasing of the activation phase shifter, represented by V¾, is an important degree of freedom for determining its nonlinear response. A representative selection of electrical biasing is considered in the example illustrated in Figure 4. The left column of Figure 4 plots the output signal amplitude as a function of the input signal amplitude
in Eq. 8, while the right column plots the transmission coefficient i.e. , a quantity
which is more commonly used in optics than machine learning. The first two rows 405 and 410, corresponding to = I.Op and 0.85p, exhibit a response which is comparable to the ReLU acti- vation function: transmission is low for small input values and high for large input values. For the bias of f& = 0.85p in the second row 410, transmission at low input values is slightly increased with respect to the response where Unlike the ideal ReLU response, the activation
at fί, = 0.85p is not entirely monotonic because transmission first goes to zero before increasing. On the other hand, the responses shown in the bottom two rows 415 and 420, corresponding to f& = 0.0p and 0.50p, are quite different. These configurations demonstrate a saturating response in which the output is suppressed for higher input values but enhanced for lower input values. Rows 415 and 420 correspond to a clipped response, with high transmission for inputs with small amplitude and reduced transmission for inputs with larger amplitude. All of the responses shown in this figure have assumed a = 0.1, which limits the maximum transmission to 1— a = 0.9.
[0045] In a variety of embodiments, by having electrical control over the activation response, its electrical bias can be connected to the same control circuitry which programs the linear interferom- eter meshes. In doing so, a single ONN hardware unit in accordance with a number of embodiments of the invention can be reprogrammed to synthesize many different activation function responses. In certain embodiments, an activation function response can be heuristically selected. Activation biases in accordance with several embodiments of the invention can be directly optimized using a training algorithm. This realization of a flexible optical-to-optical nonlinearity can allow ONNs to be applied to much broader classes of machine learning tasks.
[0046] Figure 4 shows only the amplitude response of the activation function. In fact, all of these responses also introduce a nonlinear self phase modulation to the output signal. In a number of embodiments, this nonlinear self-phase modulation can be suppressed using a push-pull interfer- ometer configuration in which the generated phase shift, Df, is divided and applied with opposite sign to the top and bottom arms.
[0047] Power consumption, computational latency, and speed need to scale for an integrated ONN, which uses meshes of integrated optical interferometers to perform matrix-vector multipli- cations and the electro-optic activation function, with respect to the number of network layers, L and the input data dimension, N.
[0048] The power consumption of an ONN in accordance with various embodiments of the invention consists of contributions from (1) the programmable phase shifters inside the interferometer mesh, (2) the optical source supplying the input vectors, x¾, and (3) the active components of the activation function such as the amplifier and photodetector. Of these, the contributions of (2) and (3) pertain to the activation function.
[0049] To quantify the power consumption, consider the minimum input optical power to a single activation that triggers a nonlinear response, or the activation function threshold. The activation function threshold can be mathematically defined as
where Df| 57=0.5 is the phase shift necessary to generate a 50% change in the power transmission with respect to the transmission with null input for a given f^. In general, a lower activation thresh- old will result in a lower optical power required at the ONN input, |XQ| 2 . Equation 10 indicates that the activation threshold can be reduced via a small V n and a large optical-to-electrical conversion gain, Gift. A chart of the relationship between optical-to-electrical gain G and the modulator V n for activation thresholds of 0.1 mW, 1.0 mW, and 10.0 mW is shown in Figure 5 for a fixed pho- todetector responsivity 9¾ = 1 A/W. Additionally, Figure 5 makes a conservative assumption that which has the highest threshold of the activation function bases shown in Figure 4.
[0050] Using the lowest activation threshold of 0.1 mW in Figure 5, the optical source to the ONN would then need to supply N · 0.1 mW of optical power. The power consumption of integrated optical receiver amplifiers varies considerably, ranging from as low as 10 mW to as high as 150 mW, depending on a variety of factors. A conservative estimate of the power consumption from a single optical-to-electical conversion circuit is L N· 100 mW. For an ONN with N = 100, the power consumption per layer from the activation function would be 10 W and would require a total optical input power of N- P± = 100 - 0.1 mW = 10 mW. Reducing the optical signal power level in accordance with several embodiments of the invention has the added advantage that it can mitigate the undesired effects of concentrated high-power regions within the interferometer mesh where many optical signals constructively interfere.
[0051] For a feedforward neural network architecture, latency can be defined by the elapsed time between supplying an input vector, L¾ and reading out its corresponding prediction vector, x^. In an integrated ONN, this delay can simply be the travel time for an optical pulse through all L-layers. In some embodiments, the propagation distance in the interferometer mesh is Z¼ = N · DMZI. where £>MZI is the length of each MZI within the mesh. In the nonlinear activation layer, the propagation length can be dominated by the delay line required to match the optical and electrical delays, and is given by
where the group velocity is the speed of optical pulses in the waveguide. Therefore,
the mathematical expression for the latency is
This indicates that the latency contribution from the interferometer mesh scales with the product LN. On the other hand, the activation function adds to the latency independently of N because each circuit is applied in parallel to all vector elements. [0052] For concreteness, assume Z½zi = 100 /mi and n Assuming that no nonlinear electrical signal conditioner is used in the activation function, Typical group delays for
integrated transimpedance amplifiers used in optical receivers can range from to 100 ps. Moreover, assuming an RC-limitcd phase modulator speed of 50 GHz yields There- fore, assuming a conservative value of ps, a network dimension of N « 100 would have a latency of 237 ps per layer, with equal contributions from the mesh and the activation function. For a ten layer network ( L = 10) the total latency would be « 2.4 ns, still orders of magnitude lower than the latency typically associated with GPUs.
[0053] The speed, or computational capacity, of an ONN can be determined by the number of input vectors, xo which can be processed per unit time. Although activation functions in accor- dance with some embodiments of the invention are not fully optical, there is no speed degradation compared to a linear ONN consisting of only interferometer meshes. The reason for this is that a fully integrated ONN would also include high-speed modulators and detectors on-chip to perform fast modulation and detection of sequences of XQ vectors and XL vectors, respectively. In a number of embodiments, the same high-speed detector and modulator elements could also be in- legrated between the linear network layers to provide the optical-electrical and electrical-optical transduction for the activation function. Similarly, the transimpedance amplifier and any other electronic components could be co-integrated with the photonic components in accordance with several embodiments of the invention. State of the art integrated transimpedance amplifiers can already operate at speeds comparable to the optical modulator and detector rates, which are on the order of 50 - 100 GHz, and thus would not be limiting factor, assuming a conservative photodetec- tor and modulator rate of 10 GHz results in an effective speed which scales as 0.01 N 2 L TFLOPS. Thus, a one layer ONN with N = 10 would perform at 1 TFLOPS, while increasing the number of inputs to N = 100 would result in a performance of 100 TFLOPS, orders of magnitude greater than the peak performance obtainable with modem GPUs.
[0054] Activation function circuits in accordance with various embodiments of the invention can be modified to remove the matched optical delay line. This modification may be advantageous for reducing the footprint of the activation and would result in · However, this can result in
a reduction of the ONN speed, which would then be limited by the combined activation delay of all L nonlinear layers in the network,
[0055] In this section, the self-phase modulation response of the electro-optic activation function is compared to an all-optical self-phase modulation achieved with the Kerr effect. The Kerr effect is a third-order optical nonlinearity which yields a change in the refractive index proportional to the local intensity. Unlike the self-phase modulation in the electro-optic activation function, the Kerr effect is lossless and has no latency. The strength of the Kerr effect inside a waveguide can be quantified through the amount of nonlinear phase shift it generates per unit input power per unit length. Mathematically, this figure of merit is defined as
where «2 is the nonlinear refractive index of the material and A is the effective mode area. Values of GK MG range from 100 (W-m) 1 in chalcogenide to 350 (W-m) 1 in silicon. For comparison, an equivalent figure of merit for the electro-optic feedforward scheme can be mathematically defined as
where V n L is the phase modulator figure of merit. A comparison of Eq. 13 and Eq. 14 indicates that while the strength of the Kerr effect is largely fixed by waveguide design and material choice, the electro-optic scheme has several degrees of freedom which allow it to potentially achieve a stronger nonlinear response.
[0056] The first design parameter is the amount of power tapped off to the photodetector, which can be increased to generate a larger voltage at the phase modulator. However, increasing a also increases the linear signal loss through the activation which does not contribute to the nonlinear mapping between the input and output of the ONN. Therefore, a should be minimized such that the optical power routed to the photodetector is large enough to be above the noise equivalent power level.
[0057] On the other hand, the product 9 \G determines the conversion efficiency of the detected optical power into an electrical voltage. Charts of nonlinear parameter GEO for the electro-optic activation as a function of gain and modulator are illustrated in Figure 6. The first chart 610 of Figure 6 compares GEO 612 to Gc b tt from silicon 614 for a = 0.50, 0.10, and 0.01, as a function of G. The responsivily is fixed at Tapping out 10% of the optical power requires a gain
of 20 dBQ to achieve a nonlinear phase shift equivalent to that of a silicon waveguide where A = 0.05 /zm 2 for the same amount of input power. Tapping out only 1 % of the optical power requires an additional 10 <1BW of gain to maintain equivalence. The gain range considered in the first chart 610 is well within the regime of what has been demonstrated in integrated transimpedance ampli- fiers for optical receivers. In fact, many of these systems have demonstrated much higher gain. In the first chart 610, the phase modulator V n L was fixed at 20 V mm. However, because a lower V n L translates into an increased phase shift for an applied voltage, this parameter can also be used to enhance the nonlinearity in accordance with some embodiments of the invention. The second chart 620 demonstrates the effect of changing the V n L for several values of of G, again, with a fixed responsivity 9¾ = 1.0 A/W. This demonstrates that with a reasonable level of gain and phase modulator performance, the electro-optic activation function can trade off an increase in latency for a significantly lower optical activation threshold than the Kerr effect.
[0058] In this section, electro-optic activation functions in accordance with several embodiments of the invention are applied to example machine learning tasks, including an exclusive-OR (XOR) function and handwritten number classification.
[0059] An exclusive-OR (XOR) is a logic function which takes two inputs and produces a single output. The output is high if only one of the two inputs is high, and low for all other pos- sible input combinations. In this example, a multi-input XOR takes N input values, given by xi . ..XN, and produces a single output value, y. The input-output relationship of the multi-input XOR function is a generalization of the two-input XOR. For example, defining logical high and low values as 1 and 0, respectively, a four-input XOR would have x and
The XOR function requires a non-trivial level of nonlinearity, meaning
that it could not be implemented in an ONN consisting of only linear interferometer meshes.
[0060] In this example, the ONN includes L layers, with each layer constructed from an iV x iV unitary interferometer mesh followed by an array of N parallel electro-optic activation functions. After the final layer, the lower N— 1 outputs are dropped to produce a single output value which corresponds to y. Unlike the ideal XOR input-output relationship described above, for the XOR task learned by the ONN, the input vectors are normalized such that they always have an Li norm of 1. This constraint is equivalent to enforcing a constant input power to the network. Additionally, because the activation function causes the optical power level to be attenuated at each layer, the high output state is normalized to be a value of 0.2. The low output value remains fixed at a value of 0.0. In several embodiments, additional ports are added with fixed power biases to increase the total input power to the network.
[0061] Charts of an XOR example are illustrated in Figure 7. The first chart 705 of Figure 7 illustrates the four-inpul XOR input-output relationship which was learned by a two-layer ONN. In this example, the electro-optic activation functions were configured to have a gain of g = 1.75p and biasing phase of f* = p. This biasing phase configuration corresponds to a ReLU-like response described above. The black markers (X) indicate the desired output values while the red circles in dicate the output learned by the two-layer ONN. The first chart 705 indicates excellent agreement between the learned output and the desired output. The evolution of the mean squared error (MSE) between the ONN output and the desired output during training confirms this agreement, as shown in the second chart 710, with a final MSE below 10 -5 .
[0062] To train the ONN, a total of 2 N = 16 training examples were used, corresponding to all possible binary input combinations along the x-axis of the first chart 705. All 16 training examples were fed through the network in a batch to calculate the mean squared error (MSE) loss function. The gradient of the loss function with respect to each phase shifter was computed by backpropa- gating the error signal through the network to calculate the loss sensitivity at each phase shifter. The above steps were repeated until the MSE converged, as shown in the second chart 710.
[0063] To demonstrate that the nonlinearity provided by the electro-optic activation function is essential for the ONN to successfully learn the XOR, the third chart 715 plots the final MSE after
5000 training epochs, averaged over 20 independent training runs, as a function of the activation function gain, g$. The shaded regions indicate the minimum and maximum range of the final MSE over the 20 training runs.
[0064] The blue curve 726 in the third chart 715, which corresponds to the ReLU-like activation, shows a clear improvement in the final MSE with an increase in the nonlinearity strength. For very high nonlinearity, above = 1.5p, the range between the minimum and maximum final MSE broadens and the mean final MSE increases. However, the best case (minimum) final MSE con- tinues to decrease, as indicated by the lower border of the shaded blue region. This trend indicates that although increasing nonlinearity allows the ONN to better learn the XOR function, very high levels of nonlinearity may also cause convergence issues in training algorithm.
[0065] A trend of decreasing MSE with increasing nonlinearity is also observed for the activation corresponding to the green curve 720 in the third chart 715. However, the range of MSE values begins to broaden at a lower value of = 1 ,Op. Such broadening may be a result of the changing slope in the activation function output. For some of the activation functions corresponding to the red (722) and orange (724) curves in the third chart 715, the final MSE decreases somewhat with an increase in g$, but generally remains much higher than the other two activation function responses.
[0066] In another example, the activation function is used to classify images of handwritten dig- its from the MNIST dataset, which has become a standard benchmark problem for ANNs. The dataset consists of 70,000 grayscale 28x28 pixel images of handwritten digits between 0 and 9.
[0067] To reduce the number of input parameters, and hence the size of the neural network, processes in accordance with a number of embodiments of the invention use a preprocessing step to convert the images into a Fourier-space representation. A 2D Fourier transform of the images can be defined mathematically as where g(m,n) is the gray scale value of the pixel at location within the image. Charts of a handwritten number recognition example are illustrated in Figure 8. The amplitudes 810 of the Fourier coefficients are shown below their corresponding images 805 in Figure 8. These coefficients are
generally complex- valued, but because the real-space map g(m,n ) is real- valued, the condition
[0068] The Fourier-space profiles are mostly concentrated around small k x and k y , corresponding to the center region of the profiles in Figure 8. This is due to the slowly varying spatial features in the images. Most of the information is carried by the small k Fourier components, and with the goal of decreasing the input size, the data can be restricted to N coefficients with the smallest
An additional advantage of this preprocessing step in accordance with many em
bodiments of the invention is that it reduces the computational resources required to perform the training process because the neural network dimension does not need to accommodate all 28 2 pixel values as inputs.
[0069] Fourier preprocessing can be particularly relevant for ONNs for two reasons. First, the Fourier transform has a straightforward implementation in the optical domain using a lens and a spatial filter for selecting the desired components. Second, this approach can take advantage of the fact that ONNs are complex-valued functions. That is to say, the N complex-valued coefficients contain 2 N degrees of freedom which could only be handled by a real-valued neural net-
work using a twice larger dimension. The ONN architecture 815 used in this example is shown schematically in Figure 8. The N Fourier coefficients close to k x = k y = 0 are fed into an optical neural network consisting of L layers, after which a drop-mask reduces the final output to 10 com- ponents which are used for one-hot encoding the digits from 0 to 9. The loss function is defined as the cross-entropy between the intensity of the output vector and the correct one-hot vector. In various embodiments, this final transformation could be performed directly by an array of photo- diodes.
[0070] During each training epoch a subset of 60,000 images from the dataset were fed through the network in batches of 500. The remaining 10,000 image-label pairs were used for validation. Charts 910 and 920 of Figure 9 illustrate comparisons between the validation accuracy and the loss function value during training for a two-layer network with N = 16 input Fourier components. The blue curves 912 correspond to an ONN with no activation function while the orange curves 914 correspond to an ONN with an electro-optic activation function configured with = 0.05p and f¾ = I.OOp. A confusion matrix for the trained ONN with an electro-optic activation function is illustrated in Figure 10. The gain setting in particular was selected heuristically. The nonlinear activation function can result in a significant improvement to the ONN performance during and af- ter training. The final validation accuracy for the ONN with the activation function is 93%, which amounts to an 8% difference as compared to the linear ONN which achieved an accuracy of 85%.
[0071] The confusion matrix 1000 corresponds to the validation data feed through the ONN with activation functions. The predicted accuracy of 93% is high considering that only N = 16 complex Fourier components were used, and the network is parameterized by only 2 x N 2 = 512 free parameters. This is comparable to the performance of a fully-connected linear classifier which takes all real-space bits as inputs and has 4010 free parameters and a validation accuracy of 92.6%. Finally, the table below shows a summary of validation accuracies after 200 epochs for an ONN without activations, with activations, and with trained activations. For file trained activations, the gain, g$, of each layer was optimized. The table shows that the accuracy can be further improved by includ- ing a third layer in the ONN and by making the activation function gain a trainable parameter. This brings the validation accuracy to 94%.
[0072] In some embodiments, an architecture for synthesizing optical-to-optical nonlinearities and its use in a feed forward ONN is provided. Rather than using optical nonlinearities, activa- tion architectures in accordance with numerous embodiments of the invention can use intermediate signal pathways in the electrical domain which can be accessed via photodetectors and phase mod- ulators. Specifically, in several embodiments, a small portion of the optical input power can be tapped out which undergoes analog processing before modulating the remaining portion of the same optical signal. Whereas all-optical nonlinearities have largely fixed responses, a benefit of the electro-optic approach demonstrated here is that signal amplification in the electronic domain can overcome the need for high optical signal powers to achieve a significantly lower activation threshold. For example, a phase modulator V n of 10 V and an optical-to-electrical conversion gain of 57 dBW, both of which are experimentally feasible, result in an optical activation threshold of
0.1 mW.
[0073] Activation function architectures in accordance with a number of embodiments of the invention can utilize the same integrated photodetector and modulator technologies as the input and output layers of a fully-integrated ONN. An ONN using this activation can suffer no reduction in processing speed, despite using analog electrical components. While there is potentially an in- crease in latency due to the electro-optic conversion process, an ONN with dimension N = 100 has approximately equal contributions to its total latency from propagation of optical pulses through the interferometer mesh as from the electro-optic activation function. This latency amounts to 2.4 ns per layer.
[0074] In many embodiments, the majority of the signal power is always transferred in the optical domain. This can eliminate the requirement of having a new optical source at each nonlinear layer of the network, as is required in previously demonstrated electro-optic neuromorphic hardware and reservoir computing architectures. Additionally, each activation function in accordance with certain embodiments of the invention is a standalone analog circuit and therefore can be applied in parallel. Finally, while many of the examples have been described to implement an architecture as an activation function in a feedforward ONN, the synthesis of low-threshold optical nonlinearlities using this circuit could be of broader interest to a number of different fields, including (but not limited to) optical computing and microwave photonics.
[0075] Although the present invention has been described in certain specific aspects, many ad- ditional modifications and variations would be apparent to those skilled in the art. It is therefore to be understood that the present invention may be practiced otherwise than specifically described. Thus, embodiments of the present invention should be considered in all respects as illustrative and not restrictive.
Previous Patent: TRAINING OF PHOTONIC NEURAL NETWORKS THROUGH IN SITU BACKPROPAGATION
Next Patent: T CELL RECEPTORS TARGETING PIK3CA MUTATIONS AND USES THEREOF