粒子群優(yōu)化算法及其java實(shí)現(xiàn)

憋了兩周終于把開題報(bào)告憋出來了,再一次證明自己不適合搞學(xué)術(shù),哎……罢艾,花了點(diǎn)時(shí)間把報(bào)告中提到的粒子群算法看了看,看了些資料嫁乘,用java跑起來昆婿。

算法簡(jiǎn)介

粒子群算法最先由Barnhart博士和Kennedy博士于1995 年提出,是一種源于對(duì)鳥群捕食行為的研究而發(fā)明的進(jìn)化計(jì)算技術(shù)蜓斧,原理是模仿鳥群尋覓食物的搜索過程仓蛆,設(shè)想鳥群在一定區(qū)域搜尋食物,在不知道食物確切位置的情況下挎春,鳥群依靠群體中個(gè)體判斷距離食物的遠(yuǎn)近程度來調(diào)節(jié)飛行方向和飛行速度看疙,最終通過群體的經(jīng)驗(yàn)和自身記憶的智慧找到食物。

算法原理

算法描述
image

image
算法流程圖
image

算法的實(shí)現(xiàn)(java)

  • Particle.java文件
package com.jiajia.pso;

import java.util.Random;

/**
 * @ClassName: Particle
 * @Author: fanjiajia
 * @Date: 2019/5/13 上午11:01
 * @Version: 1.0
 * @Description:
 */
public class Particle {

    //維數(shù)
    public  int dimension = 2;

    //粒子的位置
    public double[] X = new double[dimension];

    //局部最好位置
    public double[] pbest = new double[dimension];

    //粒子的速度
    public double[] V = new double[dimension];

    //最大速度
    public double Vmax = 2;

    //適應(yīng)值
    public double fitness;

    /**
     * 根據(jù)當(dāng)前位置計(jì)算適應(yīng)值
     * @return newFitness
     */
    public double calculateFitness() {
        //1.Ackley's function:
        //double newFitness = -20*Math.pow(Math.E,(-0.2*Math.sqrt(0.5*(X[0]*X[0]+X[1]*X[1]))))-Math.pow(Math.E,(0.5*(Math.cos(2*Math.PI*X[0])+Math.cos(2*Math.PI*X[1]))))+Math.E+20;

        //2.Sphere function
        //double newFitness = X[0]*X[0]+X[1]*X[1];

        //3.Rosenbrock function
        double newFitness = 100*(Math.pow((X[1]-X[0]*X[0]),2))+Math.pow((X[0]-1), 2);

        return newFitness;
    }


    /**
     * 初始化自己的位置和pbest
     */
    public void initialX() {
        for(int i=0;i<dimension;i++) {
            X[i] = new Random().nextInt(50);
            pbest[i] = X[i];
        }
    }
    /**
     * 初始化自己的速度
     */
    public void initialV() {
        for(int i=0;i<dimension;i++) {
            double tmp = new Random().nextDouble();//隨機(jī)產(chǎn)生一個(gè)0~1的隨機(jī)小數(shù)
            V[i] = tmp*4+(-2);
        }
    }
}
  • PSO.java
package com.jiajia.pso;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;


import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

/**
 * @ClassName: PSO
 * @Author: fanjiajia
 * @Date: 2019/5/13 上午11:02
 * @Version: 1.0
 * @Description:
 */
public class PSO {

    private static double[] gbest;//全局最優(yōu)位置

    private static double gbest_fitness = Double.MAX_VALUE;//全局最優(yōu)位置對(duì)應(yīng)的fitness

    private static int particle_num = 20;//粒子數(shù)

    private static int N = 500;//迭代次數(shù)

    private static int c1,c2 = 2;

    private static double w = 1.4;//慣性因子

    private static List<Particle> particles = new ArrayList<Particle>();//粒子群

    private static List<Double> fittessList = new ArrayList<>(N);

    /**
     * 主程序入口
     * @param args
     */
    public static void main(String[] args) {
        process();
    }

    /**
     * 初始化所有粒子
     */
    public static void initialParticles() {
        for(int i=0;i<particle_num;i++) {
            Particle particle = new Particle();
            particle.initialX();
            particle.initialV();
            particle.fitness = particle.calculateFitness();
            particles.add(particle);
        }
    }

    /**
     * update gbest
     */
    public static void updateGbest() {
        double fitness = Double.MAX_VALUE;
        int index = 0;
        for(int i=0;i<particle_num;i++) { // 找到群體中適應(yīng)值最小的粒子
            if(particles.get(i).fitness<fitness) {
                index = i;
                fitness = particles.get(i).fitness;
            }
        }
        if(fitness<gbest_fitness) { // 如果個(gè)體適應(yīng)值小于全局適應(yīng)值直奋,更新全局的最優(yōu)值為個(gè)體最優(yōu)值
            gbest = particles.get(index).pbest.clone();
            gbest_fitness = fitness;
        }
    }

    /**
     * 跟新每個(gè)粒子的速度
     */
    public static void updateV(int n) {
        for(Particle particle:particles) {
            for(int i=0;i<particle.dimension;i++) {
                double v =(0.9 - n*(0.9-0.4)/N) * particle.V[i]+c1*rand()*(particle.pbest[i]-particle.X[i])+c2*rand()*(gbest[i]-particle.X[i]);
                if(v>particle.Vmax) // 判斷速度是否超過最大的速度
                    v = particle.Vmax;
                else if(v<-particle.Vmax) // 比最大速度的相反數(shù)小
                    v = -particle.Vmax;
                particle.V[i] = v;//更新Vi
            }
        }
    }

    /**
     * 更新每個(gè)粒子的位置和pbest
     */
    public static void updateX() {
        for(Particle particle:particles) {
            for(int i=0;i<particle.dimension;i++) {
                particle.X[i] = particle.X[i] + particle.V[i];
            }
            double newFitness = particle.calculateFitness();//新的適應(yīng)值
            //如果新的適應(yīng)值比原來的小則跟新fitness和pbest
            if(newFitness<particle.fitness) {
                particle.pbest = particle.X.clone();
                particle.fitness = newFitness;
            }
        }
    }

    /**
     * 算法主要流程
     */
    public static void process() {
        int n = 0;
        initialParticles();
        updateGbest();
        while(n++<N) {
            updateV(n);
            updateX();
            updateGbest();
            fittessList.add(gbest_fitness);
            System.out.println(n+".當(dāng)前gbest:("+gbest[0]+","+gbest[1]+")  fitness="+gbest_fitness);
        }
        write2File();
    }
    
    /**
     * 返回一個(gè)0~1的隨機(jī)數(shù)
     * @return
     */
    public static double rand() {
        return new Random().nextDouble();
    }
}

代碼參考了其他資料能庆,后面有說明,但是對(duì)其中部分進(jìn)行了改進(jìn)脚线。

Particle(粒子類)中設(shè)定了三個(gè)適應(yīng)函數(shù)搁胆,Ackley,Sphere,Rosenbrock,關(guān)于這三個(gè)函數(shù)的介紹可以參考資料,這里面列出來了很多優(yōu)化測(cè)試函數(shù)邮绿,很多的paper在設(shè)計(jì)了優(yōu)化策略或者改進(jìn)相應(yīng)的優(yōu)化策略之后渠旁,都會(huì)利用其中的函數(shù)進(jìn)行測(cè)試。
這里用到的函數(shù)是Rosenbrock:

Rosenbrock函數(shù)

可以看出這里最小值在(1船逮,1顾腊,,挖胃,1)處取的杂靶。
為了看到相應(yīng)的效果梆惯,這里將全局適應(yīng)值寫到txt文件中,并利用python繪制出來(莫名的感覺繁瑣吗垮,要是python垛吗,哪有這么麻煩,只可惜最后的實(shí)驗(yàn)都是java寫)抱既。
w為1時(shí)

上面是
w
為1時(shí)职烧,即慣性系數(shù)為1時(shí)的收斂結(jié)果,可以看出防泵,算法前期搜索很快,后期較慢蝗敢,且偶爾會(huì)陷入局部最優(yōu)解里面捷泞。

慣性因子w的優(yōu)化

慣性因子w代表受上一次粒子速度的影響程度,w越大寿谴,收斂越快锁右,但容易錯(cuò)過最優(yōu)解。w越小讶泰,收斂較慢咏瑟,容易陷入局部最優(yōu)解,出不來痪署。因此改進(jìn)w成為很多改進(jìn)的焦點(diǎn)码泞,其中采用較多的是Shi建議的線性遞減權(quán)值策略,通常將w設(shè)定在[0.4狼犯,0.9]之間:

image

采用線性遞減權(quán)值策略后得到的收斂效果:
image

可以看出前期收斂直線下降余寥,且不容易陷入局部最優(yōu),最后達(dá)到全局收斂悯森。
除了上面的線性遞減權(quán)值策略宋舷,還有自適應(yīng)權(quán)值策略,隨機(jī)權(quán)重策略瓢姻。詳見參考資料[4];

參考資料

最后

生命不息祝蝠,生活好難!;眉睢绎狭!

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市收班,隨后出現(xiàn)的幾起案子坟岔,更是在濱河造成了極大的恐慌,老刑警劉巖摔桦,帶你破解...
    沈念sama閱讀 218,755評(píng)論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件社付,死亡現(xiàn)場(chǎng)離奇詭異承疲,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)鸥咖,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,305評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門燕鸽,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人啼辣,你說我怎么就攤上這事啊研。” “怎么了鸥拧?”我有些...
    開封第一講書人閱讀 165,138評(píng)論 0 355
  • 文/不壞的土叔 我叫張陵党远,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我富弦,道長(zhǎng)沟娱,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,791評(píng)論 1 295
  • 正文 為了忘掉前任腕柜,我火速辦了婚禮济似,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘盏缤。我一直安慰自己砰蠢,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,794評(píng)論 6 392
  • 文/花漫 我一把揭開白布唉铜。 她就那樣靜靜地躺著台舱,像睡著了一般。 火紅的嫁衣襯著肌膚如雪打毛。 梳的紋絲不亂的頭發(fā)上柿赊,一...
    開封第一講書人閱讀 51,631評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音幻枉,去河邊找鬼碰声。 笑死,一個(gè)胖子當(dāng)著我的面吹牛熬甫,可吹牛的內(nèi)容都是我干的胰挑。 我是一名探鬼主播,決...
    沈念sama閱讀 40,362評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼椿肩,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼瞻颂!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起郑象,我...
    開封第一講書人閱讀 39,264評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤贡这,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后厂榛,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體盖矫,經(jīng)...
    沈念sama閱讀 45,724評(píng)論 1 315
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡丽惭,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,900評(píng)論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了辈双。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片责掏。...
    茶點(diǎn)故事閱讀 40,040評(píng)論 1 350
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖湃望,靈堂內(nèi)的尸體忽然破棺而出换衬,到底是詐尸還是另有隱情,我是刑警寧澤证芭,帶...
    沈念sama閱讀 35,742評(píng)論 5 346
  • 正文 年R本政府宣布瞳浦,位于F島的核電站,受9級(jí)特大地震影響檩帐,放射性物質(zhì)發(fā)生泄漏术幔。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,364評(píng)論 3 330
  • 文/蒙蒙 一湃密、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧四敞,春花似錦泛源、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,944評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至铺厨,卻和暖如春缎玫,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背解滓。 一陣腳步聲響...
    開封第一講書人閱讀 33,060評(píng)論 1 270
  • 我被黑心中介騙來泰國(guó)打工赃磨, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人洼裤。 一個(gè)月前我還...
    沈念sama閱讀 48,247評(píng)論 3 371
  • 正文 我出身青樓邻辉,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親腮鞍。 傳聞我的和親對(duì)象是個(gè)殘疾皇子值骇,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,979評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容