c++ - 在 C++ 中计算 log-sum-exp 函数

Question

c++标准库中是否有计算指数总和对数的函数？如果没有，我应该如何编写自己的？您是否有任何未在此 wiki 文章中提及的建议？

我特别关心总和将下溢的可能性。这是您对绝对值较大的负数求幂的情况。我正在使用 c++11。

Answer 1

（shuvro 代码的 C++11 变体，根据问题使用标准库。）

template <typename Iter>
std::iterator_traits<Iter>::value_type
log_sum_exp(Iter begin, Iter end)
{
  using VT = std::iterator_traits<Iter>::value_type{};
  if (begin==end) return VT{};
  using std::exp;
  using std::log;
  auto max_elem = *std::max_element(begin, end);
  auto sum = std::accumulate(begin, end, VT{}, 
     [max_elem](VT a, VT b) { return a + exp(b - max_elem); });
  return max_elem + log(sum);
}

这个版本更通用——它适用于任何类型的值，在任何类型的容器中，只要它有相关的运算符。特别是，除非值类型有自己的重载，否则它将使用std::expand 。std::log

为了对下溢非常健壮，即使对于未知的数字类型，对值进行排序也可能是有益的。如果您对输入进行排序，第一个值将是max_elem，因此 的第一项sum可能exp(VT{0}是VT{1}。这显然没有下溢。

Answer 2

如果你想要更小的代码，这个实现可以完成这项工作：

double log_sum_exp(double arr[], int count) 
{
   if(count > 0 ){
      double maxVal = arr[0];
      double sum = 0;

      for (int i = 1 ; i < count ; i++){
         if (arr[i] > maxVal){
            maxVal = arr[i];
         }
      }

      for (int i = 0; i < count ; i++){
         sum += exp(arr[i] - maxVal);
      }
      return log(sum) + maxVal;

   }
   else
   {
      return 0.0;
   }
}

您可以在Takeda 25 的博客（日语）上看到更强大的实现（或者，通过谷歌翻译查看英语）。