R中积极支持的函数的卷积

时间:2014-05-23 03:07:34

标签: r integration convolution numerical

我想要对[0,Inf)上定义的两个函数进行卷积,比如说

f=function(x)
    (1+0.5*cos(2*pi*x))*(x>=0)

g=function(x)
    exp(-2*x)*(x>0)

使用R的集成功能我可以这样做,

cfg=function(x)
    integrate(function(y) f(y)*g(x-y),0,x)$value

通过搜索网络,似乎有更有效(更准确)的方法(比如使用fft()convolve())。任何有这种经历的人都可以解释一下吗? 谢谢!

1 个答案:

答案 0 :(得分:2)

convolvefft解决方案是获取离散结果,而不是您在cfg中定义的函数。他们可以在一些常规的离散输入上为cfg提供数值解法。

fft仅用于定期函数(仅限),因此无法提供帮助。但是,convolve具有一种称为" open"的操作模式,它模拟cfg正在执行的操作。

请注意,对于type="open",您必须反转第二个序列(请参阅?convolve,"详细信息")。您还必须仅使用结果的前半部分。以下是由c(2,3,5)执行的c(7,11,13)convolve(c(2,3,5), rev(c(7,11,13)), type='open')的卷积结果的图片示例:

             2  3  5       2  3  5   2  3  5    2  3  5      2  3  5
      13 11  7         13 11  7     13 11  7      13 11  7        13  11  7
 Sum:       14              43         94           94            65

请注意,评估前三个元素与您的集成结果类似。最后三个将用于反向卷积。

这是与您的功能的比较。你的函数,矢量化,用

绘制
y <- seq(0,10,by=.01)
plot(y, Vectorize(cfg)(y), type='l')

enter image description here

使用以下代码绘制convolve的应用程序。请注意,y每单位间隔有100个点,因此100除以合适。

plot(y, convolve(f(y), rev(g(y)), type='open')[1:1001]/100, type='l')

enter image description here

这些并不完全一致,但卷积速度要快得多:

max(abs(Vectorize(cfg)(y) - convolve(f(y), rev(g(y)), type='open')[1:1001]/100))
## [1] 0.007474999

benchmark(Vectorize(cfg)(y), convolve(f(y), rev(g(y)), type='open')[1:1001]/100, columns=c('test', 'elapsed', 'relative'))
##                                                   test elapsed relative
## 2 convolve(f(y), rev(g(y)), type = "open")[1:1001]/100   0.056        1
## 1                                    Vectorize(cfg)(y)   5.824      104