[mlpack-git] master: Add implementation of the softsign activation function. (869894f)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Thu Mar 5 22:08:44 EST 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/904762495c039e345beba14c1142fd719b3bd50e...f94823c800ad6f7266995c700b1b630d5ffdcf40

>---------------------------------------------------------------

commit 869894f985a9f7480a4e495704760cca4ffeeba1
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date:   Wed Dec 31 21:53:20 2014 +0100

    Add implementation of the softsign activation function.


>---------------------------------------------------------------

869894f985a9f7480a4e495704760cca4ffeeba1
 .../ann/activation_functions/softsign_function.hpp | 125 +++++++++++++++++++++
 1 file changed, 125 insertions(+)

diff --git a/src/mlpack/methods/ann/activation_functions/softsign_function.hpp b/src/mlpack/methods/ann/activation_functions/softsign_function.hpp
new file mode 100644
index 0000000..c9375a8
--- /dev/null
+++ b/src/mlpack/methods/ann/activation_functions/softsign_function.hpp
@@ -0,0 +1,125 @@
+/**
+ * @file softsign_function.hpp
+ * @author Marcus Edel
+ *
+ * Definition and implementation of the softsign function as described by 
+ * X. Glorot and Y. Bengio.
+ *
+ * For more information, see the following paper.
+ *
+ * @code
+ * @inproceedings{GlorotAISTATS2010,
+ *   title={title={Understanding the difficulty of training deep feedforward
+ *   neural networks},
+ *   author={Glorot, Xavier and Bengio, Yoshua},
+ *   booktitle={Proceedings of AISTATS 2010},
+ *   year={2010}
+ * }
+ * @endcode
+ */
+#ifndef __MLPACK_METHODS_ANN_ACTIVATION_FUNCTIONS_SOFTSIGN_FUNCTION_HPP
+#define __MLPACK_METHODS_ANN_ACTIVATION_FUNCTIONS_SOFTSIGN_FUNCTION_HPP
+
+#include <mlpack/core.hpp>
+
+namespace mlpack {
+namespace ann /** Artificial Neural Network. */ {
+
+/**
+ * The softsign function, defined by
+ *
+ * @f[
+ * f(x) &=& \frac{x}{1 + \abs{x}} \\
+ * f'(x) &=& (1 - \abs{x})^2
+ * f(x) &=& \left\{
+ *   \begin{array}{lr}
+ *     -\frac{y}{y-1} & : x > 0 \\
+ *     \frac{x}{1 + x} & : x \le 0
+ *   \end{array}
+ * \right
+ * @f]
+ */
+class SoftsignFunction
+{
+  public:
+  /**
+   * Computes the softsign function.
+   *
+   * @param x Input data.
+   * @return f(x).
+   */
+  static double fn(const double x)
+  {
+    if (x < DBL_MAX)
+      return x > -DBL_MAX ? x / (1.0 + std::abs(x)) : -1.0;
+    return 1.0;
+  }
+
+  /**
+   * Computes the softsign function.
+   *
+   * @param x Input data.
+   * @param y The resulting output activation.
+   */
+  template<typename InputVecType, typename OutputVecType>
+  static void fn(const InputVecType& x, OutputVecType& y)
+  {
+    y = x;
+    y.transform( [](double x) { return fn(x); } );
+  }
+
+  /**
+   * Computes the first derivative of the softsign function.
+   *
+   * @param y Input data.
+   * @return f'(x)
+   */
+  static double deriv(const double y)
+  {
+    return std::pow(1.0 - std::abs(y), 2);
+  }
+
+  /**
+   * Computes the first derivatives of the softsign function.
+   *
+   * @param y Input activations.
+   * @param x The resulting derivatives.
+   */
+  template<typename InputVecType, typename OutputVecType>
+  static void deriv(const InputVecType& y, OutputVecType& x)
+  {
+    x = arma::pow(1.0 - arma::abs(y), 2);
+  }
+
+  /**
+   * Computes the inverse of the softsign function.
+   *
+   * @param y Input data.
+   * @return f^{-1}(y)
+   */
+  static double inv(const double y)
+  {
+    if (y > 0)
+      return y < 1 ? -y / (y - 1) : DBL_MAX;
+    else
+      return y > -1 ? y / (1 + y) : -DBL_MAX;
+  }
+
+  /**
+   * Computes the inverse of the softsign function.
+   *
+   * @param y Input data.
+   * @param x The resulting inverse of the input data.
+   */
+  template<typename InputVecType, typename OutputVecType>
+  static void inv(const InputVecType& y, OutputVecType& x)
+  {
+    x = y;
+    x.transform( [](double y) { return inv(y); } );
+  }
+}; // class SoftsignFunction
+
+}; // namespace ann
+}; // namespace mlpack
+
+#endif



More information about the mlpack-git mailing list